[백준] 11651 좌표 정렬하기 2 Java

문제

2차원 평면 위의 점 N개가 주어진다. 좌표를 y좌표가 증가하는 순으로, y좌표가 같으면 x좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

 

입력

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

 

출력

첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

 

처음 문제 보고 바로 생각난건 배열에 저장하고 if문으로 상황을 나눠서 swap 해주는 방식이었다.

그런데 이렇게 하면 이중 for문으로 다 확인해줘야 하고 제출하면 시간초과가 난다.

그렇게 생각해 낸 두번째 방법은 Arrays.sort()로 배열을 정렬해주는 방식이었다.

 

코드

import java.io.*;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    static int[][] arr;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int N = Integer.parseInt(br.readLine());
        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());
        }

        for (int i = 0; i < N; i++) {
            for (int j = i + 1; j < N; j++) {
                if (i == j) continue;
                if (arr[i][1] > arr[j][1]) {    // y좌표가 클 때
                    swap(i, j);
                } else if (arr[i][1] == arr[j][1] && arr[i][0] > arr[j][0]) { //y좌표 같고 x좌표가 클 때
                    swap(i, j);
                }
            }
        }
        for (int k = 0; k < N; k++) {
            System.out.println(arr[k][0] + " " + arr[k][1]);
        }
    }

    static void swap(int a, int b) {
        int[] temp = arr[b];
        arr[b] = arr[a];
        arr[a] = temp;
    }

}

import java.io.*;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    static int[][] arr;

    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());
        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, (e1, e2) -> {
            if (e1[1] == e2[1]) {
                return e1[0] - e2[0];
            } else {
                return e1[1] - e2[1];
            }
        });

        for (int k = 0; k < N; k++) {
            sb.append(arr[k][0]).append(" ").append(arr[k][1]).append("\n");
        }
        System.out.println(sb);
    }
}

 

문제 출처

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