본문 바로가기

백준/정렬

백준 11650 자바

반응형

 

백준 정렬 문제이고 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