백준 - 11650번 - 좌표 정렬하기

이상훈·2023년 4월 11일
0

11650번

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

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

		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		int N = Integer.parseInt(br.readLine());

		int[][] arr = new int[N][2];

		StringTokenizer st;
		for(int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine());
			arr[i][0] = Integer.parseInt(st.nextToken());
			arr[i][1] = Integer.parseInt(st.nextToken());
		}

		Arrays.sort(arr, (e1, e2) -> {
			if(e1[0] == e2[0]) {
				return e1[1] - e2[1];
			} else {
				return e1[0] - e2[0];
			}
		});

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


}

풀이


Array.sort의 API를 살펴보면 인자로 2개의 인자를 받는 것을 알 수 있습니다.

public static <T> void sort(T[] a, Comparator<? super T> c)

의 인자 두개를 받고

Arrays.sort(arr, (e1, e2))

람다식을 통해서 재정의 해준다.
리턴값이 양이냐 음이냐를 통해서 정렬을 해준다.

Arrays.sort(arr, (e1, e2) -> {
				if(e1[0] == e2[0]) {
					return e1[1] - e2[1];
				} else {
					return e1[0] - e2[0];
				}
			});

솔직히 이 방법을 찾아보고 많이 생소했고 익숙하게 사용하기까지 더 사용해봐야겠다.

0개의 댓글