[Java] 백준 11651번 : 좌표 정렬하기 2

좌표 정렬하기 2 

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율

1 초

256 MB

95319

62652

53525

67.144%

문제

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

입력

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

출력

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

예제 1

5
0 4
1 2
1 -1
2 2
3 3

예제  2 

1 -1
1 2
2 2
3 3
0 4

 

 

 

---

 

간단한 문제이다. 

 

Arrays.sort()와 Comparator를 이용하면 금방 풀 수 있다.

 

이차원 배열 arr[N][2]이 있다. (N은 주어진 좌표 개수, 2는 x,y 좌표를 의미)

배열 arr을 Arrays.sort()를 이용하여 정렬할 때, 우리만의 정렬 기준(Comparator)을 선언해 준다.

 

우선 arr[N][2]를 열어보면 

아래와 같이 배열 안에 작은 배열이 있는 식으로 볼 수 있다.

arr = { {1, 2} ,

          {2, -1},

          {-1, 2},

          .... }

 

Comparator에 우리가 선언할 compare은 arr 안의 작은 배열의 값을 비교하여 정렬하는 것이다.

        // 핵심
        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] a, int[] b){
                if(a[1] == b[1]){ // arr[i][1] == arr[i+1][1] 두 점의 Y 좌표를 비교
                    return a[0]-b[0];  //같으면 X의 값을 비교하여 정렬
                } else { 
                    return a[1]-b[1];  //다르면 Y의 값을 비교하여 정렬
                }
            }
        });

 

 

전체 코드는 다음과 같다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Comparator;
import java.util.Arrays;

public class BOJ11651_AlignTwoCoordinate {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());

        int[][]arr = new int[n][2];
		
        // 좌표를 배열에 담기
        for(int i=0;i<n;i++){
            String[] a = br.readLine().split(" ");
            arr[i][0] = Integer.parseInt(a[0]);
            arr[i][1] = Integer.parseInt(a[1]);
        }
		
        // 핵심
        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] a, int[] b){
                if(a[1] == b[1]){
                    return a[0]-b[0];
                } else {
                    return a[1]-b[1];
                }
            }
        });

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