[알고리즘] java 좌표정렬 (compareTo)

 

java 좌표정렬

 

7. 좌표 정렬

 

설명

N개의 평면상의 좌표(x, y)가 주어지면 모든 좌표를 오름차순으로 정렬하는 프로그램을 작성하세요.

정렬기준은 먼저 x값의 의해서 정렬하고, x값이 같을 경우 y값에 의해 정렬합니다.

입력

첫째 줄에 좌표의 개수인 N(3<=N<=100,000)이 주어집니다.

두 번째 줄부터 N개의 좌표가 x, y 순으로 주어집니다. x, y값은 양수만 입력됩니다.

출력

N개의 좌표를 정렬하여 출력하세요.

예시 입력 1 

5
2 7
1 3
1 2
2 5
3 6

예시 출력 1

1 2
1 3
2 5
2 7
3 6

 

 

 

풀이:

comparable 인터페이스의 구현 클래스인 Point클래스를 생성.

compareTo메서드를 오버라이드한다.

*이때 오름차순 정렬 시 음수 값을 리턴해야 한다. ( 오름차순 정렬시 앞의 값에서 뒤의 값을 빼면 음수가 나온다)

(this 다음에 o가 온다고 할 때, 오름차순 정렬이면 this가 10, o가 20 이므로 this - o를 해서 음수 반환)

반대로 내림차순 정렬 시 양수 값을 리턴해야 한다.

 

 

import java.util.*;

class Point implements Comparable<Point>{
    public int x,y;
    Point(int x, int y){
        this.x = x;
        this.y = y;
    }
    @Override
    public int compareTo(Point o){
        if(this.x==o.x) return this.y - o.y;
        else return this.x - o.x;
    }
}


public class Main {
    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);
        int n = Integer.parseInt(sc.nextLine());

        ArrayList<Point> arr = new ArrayList<>();
        for(int i=0; i<n; i++){
            int x = sc.nextInt();
            int y = sc.nextInt();
            arr.add(new Point(x,y));
        }
        Collections.sort(arr);

        for(Point o : arr){
            System.out.println(o.x + " " + o.y);
        }
    }
}