[코딩테스트] 좌표 정렬 (이차원 배열 정렬 방식 2가지)

최지나·2024년 3월 11일
2

코딩테스트

목록 보기
129/154
post-thumbnail

문제

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 인터페이스를 구현한 객체의 compareTo 메서드를 수정하여 좌표 정렬하는 방법을 기록하고자 한다

코드

    1. 직접 정렬 알고리즘 구현
public int[][] compareCoordinate(int N, int[][] arr) {
    Arrays.sort(arr, (a, b) -> {
        int compare = Integer.compare(a[0], b[0]);
        if (compare != 0)
            return compare;
        else {
            return Integer.compare(a[1], b[1]);
        }
    });
    return arr;
}
    1. Comparable 객체의 compareTo 메서드 수정
class Point implements Comparable<Point> {
    public int x;
    public int 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; // y 오름차순. 내림차순 시 return o.y - this.y;
        else
            return this.x - o.x; // x 오름차순. 내림차순 시 return o.x - this.x;
    }
}
 ArrayList<Point> arr = new ArrayList<>();
 // ...
 Collections.sort(arr);

정리

  • 2차원 배열을 정렬 시 직접 알고리즘을 구현하면 간단하게 구현할 수 있지만, 새로 compareTo 메서드를 재정의하는 방식이 객체의 정렬 기준을 변경하고 싶을 때 클래스의 메서드만 수정하면 되므로 객체 지향적이라는 장점이 있다
  • 구조의 복잡성과 유지보수성의 trade-off를 생각해서 둘 중에 적절한 방법을 선택해 사용해야겠다!
profile
의견 나누는 것을 좋아합니다 ლ(・ヮ・ლ)

0개의 댓글