[BOJ] 11650: 좌표 정렬하기

https://www.acmicpc.net/problem/11650

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 1

2 2

3 3

3 4

 

---

[문제 풀이]

입력받은 2차원 배열의 첫 번째 원소를 기준으로 배열을 정렬하는 문제이다.

Comparator을 이용하여 배열을 정렬해야 하는데, 이는 다음과 같다.

compare 함수의 입력을 int[] o1, int[] o2라고 하고 x좌표인 첫 번째 원소를 기준으로 해야하므로 o1[0]과 o2[0] 값을 비교하여 정렬한다. 이는 return o1[0] - o2[0]을 이용하면 된다.

그러나 만약 두 값이 같다면, y좌표인 두 번째 원소값 또한 비교하여야 한다. 따라서 o1[0]==o2[0]이라면, return o1[1]-o2[1]을 이용하면 된다.

 

[코드]

import java.util.*;

public class Main {

    public static void main(String[] args) {
    
      Scanner sc = new Scanner(System.in);
      int N = sc.nextInt();
      int[][] grid = new int[N][2];
      
      for(int i=0;i<N;i++) {
       grid[i][0]=sc.nextInt();
       grid[i][1]=sc.nextInt();
      }
      
      Arrays.sort(grid, new Comparator<int[]>() {
       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++) {
       System.out.println(grid[i][0]+" "+grid[i][1]);
      }
       
    }
}