백준 | 11651 : 좌표 정렬하기 2 (Java)

문제에 대한 자세한 정보는 백준 | 11651번 : 좌표 정렬하기 2에서 확인할 수 있다.

풀이

1. 좌표를 저장할 2차원 배열(length : [n][2])을 생성한다.
2. 좌표를 입력받아 2차원 배열에 저장한다.
3. 정렬을 위한 Arrays.sort() 사용을 위해 익명 클래스로 Comparator를 생성해 compare 함수를 작성한다.

소스코드

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

public class Main {

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

		int n = Integer.parseInt(br.readLine());
		int[][] coordinate = new int[n][2];

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

		Arrays.sort(coordinate, new Comparator<int[]>() {

			@Override
			public int compare(int[] o1, int[] o2) {
				// TODO Auto-generated method stub
				if (o1[1] == o2[1])
					return o1[0] - o2[0];
				return o1[1] - o2[1];
			}
		});

		for (int i = 0; i < n; i++) {
			bw.write(String.valueOf(coordinate[i][0]) + " " + String.valueOf(coordinate[i][1]) + "\n");
		}

		br.close();
		bw.close();
	}

}

메모리, 시간

메모리 : 62692KB
시간 : 844ms

풀이 후

11650번 : 좌표 정렬하기와 정렬 기준만 다른 문제였다. 11650번을 풀이할 때는 coordinate class를 작성해 Collections.sort()를 사용했다면 이번에는 2차원 배열에 저장 후 compare 함수를 작성하는 방식으로 풀어봤다.