[JAVA] 백준 11650번 【좌표 정렬하기】

---

이 문제는, 단순히 1차원 오름차순 정렬이 아닌,

 

x좌표가 같은 경우 y좌표도 비교해야 하기 때문에, sort 함수의 정렬 조건을 변경시켜주어야 한다.

 

[JAVA] Comparator 와 Comparable에 대해 이해하기

 

마침 최근에, 한 백엔드 스쿨 동기분 덕분에 Comparator 에 대한 이해를 한 상태라서 수월하게 해결할 수 있었다.

 

먼저, Arrays.sort(int [][] array) 와 같은 방식으로 sort 메서드를 사용할 수 없기 때문에

 

뒤에 Comparator 객체를 생성하고 compare 메서드를 오버라이딩해서 정확한 정렬 기준을 제공해주어야 메서드가 사용이 가능하다.

 

참고로 좌표를 기록할 배열을 생성할 때 primitive 타입으로 만들면 안된다.

 

compare 메서드가 매개변수로서 object를 취급하기 때문이다.

 

따라서 int [][] array 가 아닌 Integer [][] array 로 생성해 주어야 한다.

 

 

 

만약 첫번째 인자로, -1 1 좌표가 들어왔다고 생각했을 때,

 

array[0][0] = -1

array[0][1] = 1  로 기록할 것이고

 

두번째 인자로 2,2 좌표가 들어왔을 때,

 

array[1][0] = 2

array[1][1] = 2 가 기록될 것이다.

 

즉 array[i][0] 은 i 번째로 입력된 좌표의 x좌표이고 

array[i][1] 은 i 번째로 입력된 좌표의 y좌표가 될 것이다.

 

 

class CustomComparator implements Comparator<Integer[]> {
	@Override
	public int compare(Integer[] a1, Integer[] a2) {
		if (a1[0].equals(a2[0])) {
			return a1[1] - a2[1];
		}
		return a1[0] - a2[0];
	}
}

여기서 중요한 부분은, 지네릭을 사용해서 Integer[] 만 들어올 수 있게 해두었다는 것이다.

 

compare 메서드에서 받는 매개변수, 즉 비교 대상도 Integer[] 로 입력받을 수 있게 된다.

 

그리고 a1[0] , 즉 어떤 좌표의 x좌표와 a2[0] , 다른 좌표의 x좌표가 같다면, 

 

a1[1] - a2[1] , 즉 두 좌표의 y좌표 값의 차이가 반환되고,

 

오름차순 정렬이므로 a1의 y좌표가 더 크다면, 양수값이 반환되어 두 객체의 위치를 바꿀 것이다.

 

조건문을 만족하지 못하면 

 

a1[0] - a2[0] 이 반환되어, 이 경우 역시 a1의 x좌표가 더 크다면 양수값이 반환되어 두 위치를 바꿀 것이다.

 

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		int N = Integer.valueOf(br.readLine());
		Integer[][] array = new Integer[N][2];
        
		
		for (int i = 0; i < N; i++) {
			String[] input = br.readLine().split(" ");
			array[i][0] = Integer.valueOf(input[0]);
			array[i][1] = Integer.valueOf(input[1]);
		}
        
        
		//array[i][0] 은 i 번째로 입력된 좌표의 x좌표이고 
        //array[i][1] 은 i 번째로 입력된 좌표의 y좌표가 될 것이다.	
		
//		System.out.println(Arrays.deepToString(array));
		Arrays.sort(array, new CustomComparator());
        //CustomComparator 객체를 불러왔다.
        
//		System.out.println(Arrays.deepToString(array));
		
		
		for (int i = 0; i < N; i++) {
			sb.append(array[i][0]).append(" ").append(array[i][1]).append("\n");
		}
		System.out.println(sb);
	}
}

class CustomComparator implements Comparator<Integer[]> {
	@Override
	public int compare(Integer[] a1, Integer[] a2) {
		if (a1[0].equals(a2[0])) {
			return a1[1] - a2[1];
		}
		return a1[0] - a2[0];
	}
}

 

다른 분들의 풀이를 봤는데, 람다식을 이용해서 해결한 사람들이 대부분이었다.

 

아직 람다식에 익숙치 못해서, 생각을 못했는데, 다음 문제부터는 람다식도 생각해봐야겠다.