기록하는 개발자

[Java] 2차원 배열에 기준을 정해 정렬하기 (백준 : 11650번) 본문

Algorithm

[Java] 2차원 배열에 기준을 정해 정렬하기 (백준 : 11650번)

밍맹030 2021. 8. 8. 00:19
728x90

https://www.acmicpc.net/problem/11650

 

11650번: 좌표 정렬하기

첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.

www.acmicpc.net

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));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;
        int n = Integer.parseInt(br.readLine());
        int[][] arr = new int[n][2];

        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, (n1, n2) -> {
            int temp = (n1[0] == n2[0])? n1[1] - n2[1] : n1[0] - n2[0];
            return temp;
        });
        
        for(int i = 0 ; i< n ; i++)
            sb.append(arr[i][0] + " " + arr[i][1]).append('\n');

        System.out.println(sb);
    }
}

좌표(arr[i][0], arr[i][1])에서 arr[i][0]이 x 좌표, arr[i][1]이 y 좌표일 때,

2차원 배열을 Arrays.sort의 함수를 람다식으로 Override 하여

좌표가 저장된 배열을 x좌표가 증가하는 순으로 우선 정렬 하고x좌표가 같은 경우 y좌표가 증가하는 순서로 정렬하도록 한다.

Arrays.sort(arr, (n1, n2) -> { // x좌표가 같은 경우 y좌표로 비교
    int temp = (n1[0] == n2[0])? n1[1]-n2[1] : n1[0]-n2[0];
    return temp;
});
728x90