반응형
백준 정렬 문제이고 2차원 배열과 Comparator로 풀 수 있습니다.
Arrays.sort(arr, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if(o1[0] == o2[0]) {
return o1[1] - o2[1];
} else {
return o1[0] - o2[0];
}
}
});
이 부분이 문제에서 주어진 대로 정렬을 하는 부분인데
o1[0], o2[0]이 x좌표 o1[1], o2[1]이 y좌표를 의미합니다.
1차원 배열을 정렬할 때는 Arrays.sort(배열 이름) 을 하면 되지만 2차원 배열을 정렬하는 것은 기준에 따라서 까다로울 수 있습니다. 저도 처음엔 많이 헤맸는데 이런 문제를 계속 풀고 풀었던 문제도 다시 풀어보면 뭔가 감이 잡히실 겁니다.
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));
StringBuilder sb = new StringBuilder();
int N = Integer.parseInt(br.readLine());
int[][] arr = new int[N][2];
for (int i = 0; i < N; i++) {
StringTokenizer st = new StringTokenizer(br.readLine());
arr[i][0] = Integer.parseInt(st.nextToken());
arr[i][1] = Integer.parseInt(st.nextToken());
}
Arrays.sort(arr, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if(o1[0] == o2[0]) {
return o1[1] - o2[1];
} else {
return o1[0] - o2[0];
}
}
});
for (int i = 0; i < N; i++) {
sb.append(arr[i][0]).append(" ").append(arr[i][1]).append('\n');
}
System.out.println(sb);
}
}
반응형
'백준 > 정렬' 카테고리의 다른 글
백준 1940 자바 (0) | 2022.08.26 |
---|---|
백준 1269 자바 (0) | 2022.08.21 |
백준 11728 자바 (0) | 2022.08.21 |
백준 1427 자바 (0) | 2022.08.20 |
백준 1181 자바 (0) | 2022.08.20 |