[백준] 11650번. 좌표 정렬하기 (Java)

[문제 바로가기] https://www.acmicpc.net/problem/11650

문제

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

입력

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

출력

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

---

풀이

1. 2차원 배열(=arr[N][2])로 x, y좌표 값 받기
2. 버블정렬로 x 좌표 비교 후, y 좌표 비교하여 swap
3. 그러나 코드가 너무 길어지고 무엇보다 백준에서 시간초과 발생

Arrays.sort()를 이용해서 2차원 배열 정렬하는 방법이 없을까 생각했다.
그리고 Arrays.sort()를 오버라이드 하는 방법을 알게 되었다.

• 우선 1차원 배열의 경우, 아래와 같이 오름차순으로 정렬한다.

Arrays.sort(arr);

• 2차원 배열의 경우, 람다식 사용하기!
  기존의 Arrays.sort()를 람다식을 통해 오버라이드 하면 된다.

• 그렇다면 람다식이란 무엇일까?
  람다식이란, 함수의 이름이 없는 함수인 람다함수를 이용한 식으로 아래와 같이 작성한다.

(x, y) -> { 함수내용; }

그럼 이제 이것을 적용한 2차원 배열의 Arrays.sort()를 작성하면..

Arrays.sort(arr, (o1, o2) -> {
    if(o1[0] == o2[0]) {
		return Integer.compare(o1[1], o2[1]);
    } else {
        return Integer.compare(o1[0], o2[0]);
    }
});

---

코드

주의사항!
JAVA 11 버전에서 실행이 된다. JAVA 8은 시간 초과로 사용 불가

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

public class Main {
	public static void main(String[] args) {

		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		
		int[][] arr = new int[N][2];
		
		for(int i = 0; i < N; i++) {
			arr[i][0] = sc.nextInt();
			arr[i][1] = sc.nextInt();
		}
		
		Arrays.sort(arr, (o1, o2) -> {
			if(o1[0] == o2[0]) {
				return Integer.compare(o1[1], o2[1]);
			} else {
				return Integer.compare(o1[0], o2[0]);
			}
		});
				
		for(int i = 0; i < N; i++) {
			System.out.println(arr[i][0] + " " + arr[i][1]);
		}
	}
}