[백준] 10814번 나이순 정렬

링크: https://www.acmicpc.net/problem/10814

문제:
온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.

입력:
첫째 줄에 온라인 저지 회원의 수 N이 주어진다. (1 ≤ N ≤ 100,000)

둘째 줄부터 N개의 줄에는 각 회원의 나이와 이름이 공백으로 구분되어 주어진다. 나이는 1보다 크거나 같으며, 200보다 작거나 같은 정수이고, 이름은 알파벳 대소문자로 이루어져 있고, 길이가 100보다 작거나 같은 문자열이다. 입력은 가입한 순서로 주어진다.

출력:
첫째 줄부터 총 N개의 줄에 걸쳐 온라인 저지 회원을 나이 순, 나이가 같으면 가입한 순으로 한 줄에 한 명씩 나이와 이름을 공백으로 구분해 출력한다.

제출:

import java.io.*;
import java.util.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		
		int n = Integer.parseInt(br.readLine());
		
		String[][] a = new String[n][2];
		for (int i=0; i<n; i++) {
			st = new StringTokenizer(br.readLine(), " ");
			a[i][0] = st.nextToken();
			a[i][1] = st.nextToken();
		}
		Arrays.sort(a, (o1, o2) -> {
				return Integer.parseInt(o1[0]) - Integer.parseInt(o2[0]);
		});
		for (int i=0; i<n; i++) {
			System.out.println(a[i][0] + " " + a[i][1]);
		}
	}
}

2차원 배열을 나이를 기준으로 오름차순 정렬하는 문제였다.
2차원 배열을 정렬하기 위해서는 Comparator 익명클래스를 구현해주어야 하는데 Java8 버전 이상에서는 간단하게 람다식으로 구현할 수 있다.

Arrays.sort(a, (o1, o2) -> {
	return Integer.parseInt(o1[0]) - Integer.parseInt(o2[0]);
});

나이와 이름을 함께 String으로 입력받았기 때문에 o1[0]과 o2[0]의 값인 나이를 숫자로 변환하여 비교한 후 오름차순으로 정렬해주었다.
만약 내림차순으로 정렬하고 싶으면 다음과 같이 return 부분에서 비교하는 앞뒤 순서를 바꾸어 주면 된다.

Arrays.sort(a, (o1, o2) -> {
	return Integer.parseInt(o2[0]) - Integer.parseInt(o1[0]);
});

Comparator의 comparing 메서드를 이용하는 방법도 있는데 아래와 같다.

// a 배열의 첫 번째 요소 기준 오름차순
Arrays.sort(a, Comparator.comparing((String[] o) -> Integer.parseInt(o[0])));
// a 배열의 첫 번째 요소 기준 내림차순
Arrays.sort(a, Comparator.comparing((String[] o) -> Integer.parseInt(o[0])).reversed());
// a 배열의 두 번째 요소 기준 오름차순
Arrays.sort(a, Comparator.comparing((String[] o) -> Integer.parseInt(o[1])));
// a 배열의 두 번째 요소 기준 내림차순
Arrays.sort(a, Comparator.comparing((String[] o) -> Integer.parseInt(o[1])).reversed());