[알고리즘 PS] 백준 11651번 좌표 정렬하기 2 자바 문제 풀이

문제

해당 포스팅은 백준의 11650번 좌표 정렬하기 의 접근과 해결 방법을 설명한 글 입니다.
정답 소스 코드를 확인하시려면 solve url 에서 확인 가능합니다.

이 문제를 해결하기 위해 어떤 방식으로 접근해야 하는지를 먼저 생각해보자.

해결법

이번 좌표 정렬하기 문제는 기본적인 정렬 문제와 조금 성격이 다른 문제이다.

보통 이런 구현 문제를 정렬 기준 변경하기, 정렬 기준 바꾸기, 정렬 기준 재정의라는 타이틀이라고 한다.

내가 경험했던 코딩테스트에서도 위와 같은 방식의 문제를 몇번 마주한 적이 있었다.

물론 이런식으로 대놓고 정렬 기준을 바꾸세요! 라고 하지는 않는다.

대충 문제를 보다가 아~ 이건 객체끼리 일반적 정렬 말고 다른 정렬을 하고 로직을 돌리면 더 빠르겠다 하는 것이 대부분이었다.

 

이 문제를 풀기 위해서는 자바의 Comparator을 알아야 한다.

 

만약 Comparator과 Comparable에 대해서 알지 못한다면 Java에서 Comparator로 정렬 기준 바꾸기에서 확인할 수 있다.

접근법

우리는 다음과 같은 과정을 거쳐서 문제를 해결할 것이다.

1. Point 클래스 생성
2. 입력 처리
3. MyComparator 구현
4. Arrays.sort
5. 출력 처리

정답 코드

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int n = Integer.parseInt(br.readLine());
        Point[] points = new Point[n];
        for (int i = 0; i < n; i++) {
            String[] xy = br.readLine().split(" ");
            points[i] = new Point(
                    Integer.parseInt(xy[0]),
                    Integer.parseInt(xy[1]));
        }

        Comparator<Point> myComparator = new Comparator<Point>() {
            @Override
            public int compare(Point p1, Point p2) {
                if(p1.y > p2.y) {
                    return 1;
                }else if (p1.y == p2.y){
                    if(p1.x > p2.x) {
                        return 1;
                    }else return -1;
                }
                return -1;
            }
        };

        Arrays.sort(points, myComparator);

        for (int i = 0; i < n; i++) {
            bw.write(points[i].x + " " + points[i].y + "\n");
        }
        bw.flush();
        bw.close();
    }
}

class Point {
    int x;
    int y;
    public Point(int x, int y) {
        this.x = x;
        this.y = y;
    }
}

---

회고

정답 소스 코드를 확인하시려면 solve url 에서 확인 가능합니다.