[백준/자바] 11650 : 좌표 정렬하기

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

11650번: 좌표 정렬하기

---

x좌표를 오름차순으로 정렬하고, x좌표가 같으면 y좌표가 증가하는 순으로 정렬하는 문제이다.

좌표를 2차원 배열에 저장하고, Arrays.sort와 Comparator의 compare 함수를 오버라이딩해 정렬 조건을 걸어주면 된다.

compare 함수는 람다함수를 사용해 더 편리하게 작성할 수 있다.

 

정렬 코드는 다음과 같다.

정렬 알고리즘 상 반환값이 양수이면 위치를 교환한다.

// p1, p2의 자료형은 int[]
Arrays.sort(arr, (p1, p2) ->{
   if(p1[0] == p2[0]){ // 두 좌표의 x값이 같으면
      return p1[1] - p2[1] // p1[1] < p2[1]
   }else{
      return p1[0] - p2[0]
   }
})

 

 

전체 코드는 다음과 같다.

import java.util.Arrays;
import java.util.Scanner;

public class Boj_11650 {

   public static void main(String[] args) {
       // TODO Auto-generated method stub
       Scanner sc = new Scanner(System.in);

       int N = sc.nextInt();
       int[][] arr = new int[N][2]; // 2차원 배열에 좌표 저장

       for(int i = 0; i<N; i++) {
           arr[i][0] = sc.nextInt();
           arr[i][1] = sc.nextInt();
       }

       Arrays.sort(arr, (e1, e2)->{
           if(e1[0] == e2[0]) { // 첫 번째 값이 같으면
               return e1[1] - e2[1]; // e1 < e2
           }
           else {
               return e1[0] - e2[0];
           }
       });

       for(int i = 0; i < N; i++) {
          System.out.println(arr[i][0] + " " + arr[i][1]);
      }
       sc.close();
   }
}

출력을 아래와 같이 사용하니 시간초과 

for(int[] arrays: arr) {
    System.out.printf("%d %d\n", arrays[0], arrays[1]);
}

---

정렬하는 문제에서 두 개 이상의 조건이 있다면

Arrays.sort()의 Comparator를 람다함수를 통해 커스텀해 조건에 맞게 사용!