[Java] 백준-11650번 "좌표 정렬하기"

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

---

입력

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

---

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

---

예제1

입력:
5
3 4
1 1
1 -1
2 2
3 3

출력: 
1 -1
1 1
2 2
3 3
3 4

---

알고리즘

java에서 Comparator라는 것을 사용하여 정렬해주면 된다.

Comparator를 사용하기 위해선

import java.util.Comparator;

를 해주어야 한다.

아래는 Comparator를 사용한 코드이다.

Arrays.sort(arr, new Comparator<int[]>() {
	@Override
    public int compare(int[] o1, int[] o2) {
    	if(o1[0] == o2[0]) return o1[1] - o2[1];
        else return o1[0] - o2[0];
    }
});

arr[i][0]과 arr[i+1][0]을 비교하여 동일하면, arr[i][1]과 arr[i+1][1]을 비교하여 정렬한다. 동일하지 않으면, 그냥 바로 정렬해주면 된다.

---

내 소스 코드

import java.util.Scanner;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
    public static void main(String[] args) {
        Scanner s = new Scanner(System.in);
        int n=0, cnt=0;
        double sum=0;
        StringBuffer sb = new StringBuffer();

        n = s.nextInt();

        int[][] arr = new int[n][2];
        for(int i=0;i<n;i++) {
          arr[i][0] = s.nextInt();
          arr[i][1] = s.nextInt();
        }

        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
              if(o1[0] == o2[0]) return o1[1] - o2[1];
              else return o1[0] - o2[0];
            }
        });

        for(int i=0;i<n;i++) {
          sb.append(arr[i][0]);
          sb.append(" ");
          sb.append(arr[i][1]);
          sb.append("\n");
        }
        System.out.print(sb);
    }
}