[JAVA] 11650번 좌표 정렬하기

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

11650번: 좌표 정렬하기

 

---

후기

 

2차원 배열까지 접근하는 건 알겠는데 도저히 정렬을 어떤 식으로 해야 되는지 감이 안 왔다. 

Arrays.sort() 메소드의 경우 기본적으로 인자로 1차원 배열만 인자로 받기 때문이다. 다행히 Compartor를 이용해서 해결할 수 있었다.

 

---

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

public class 좌표정렬하기_11650 {
    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());
        }

먼저 입력 갯수 N을 입력 받는다.

그리고 N개의 행을 가지고 2개의 열을 갖는 2차원 배열을 선언한다.

 

그리고 N번 반복하여 StringTokenizer를 생성과 동시에 생성자 인자로 br.readLine() 메소드를 호출하여 한 줄씩 입력받는다. 한 줄은 공백(" ")을 기준으로 Token화 되어 1열에는 x가 2열에는 y가 들어가도록 한다. 이때 토큰은 String 형이므로 Integer.parseInt() 메소드로 int형으로 변환해준다.

 

 

// 알고리즘 - Compator을 이용해서 객체를 비교하기
        Arrays.sort(arr,new Comparator<int[]>() {
           @Override
           public int compare(int[] e1, int[] e2) {
               if(e1[0] == e2[0]){  // x값이 같으면
                   return e1[1] - e2[1];    // y는 오름 차순으로
               }
               else {
                   return e1[0] - e2[0];    // x도 오름차순으로 정렬
               }
           }
        });

 

2차원 배열도 Arrays.sort(배열, 정렬 기준)을 통해 정렬할 수 있다. 여기서 new Compartor이 등장하는데 당황하지 말자. 

 

일반적으로 정렬을 하려면 비교 기준이 있어야 한다. 예를들어 기본자료형 int의 경우 우리는 3 > 1와 같이 부등호를 통해 비교를 할 수 있다. 하지만 2차원 배열인 int[][]은 int[]라는 자료형이 배열로 묶인 걸로 볼 수 있다. 그리고 여기서 int[]는 기본자료형이 아닌 객체이다. 객체의 경우에는 부등호를 통해 비교를 할 수 없다. 따라서 우리가 스스로 비교 기준을 만들어줘야 한다.

 

우리는 함수형 인터페이스 Compartor의 compare 메소드를 구현하여 새로운 기준을 만들 수 있다. 여기서는 새로운 클래스를 만드는 대신 익명 클래스를 사용하여 클래스 정의 + 인스턴스 생성을 동시에 하였다.

 

if 문에서 첫 번째 원소를 비교한다. 즉 x값이 같으면 y값을 비교한다. 하지만 x값이 다를 경우에는 오름차순으로 결정하게 정의하였다. 

 

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

출력은 다음과 같이 StringBuilder를 이용해 성능에 문제가 없게끔 했다.

---

출력 결과

 

 

참고

https://st-lab.tistory.com/110