[JAVA] 백준 11651번 【좌표 정렬하기 2】

---

이번 문제는, 이전에 풀었었던 11650 문제와 굉장히 흡사하다.

 

[JAVA] 백준 11650번 【좌표 정렬하기】

 

다만, 차이점이 있다면, y좌표 기준으로 먼저 오름차순 정렬을 한 뒤에, y좌표가 같다면, x좌표를 비교해서 오름차순으로 비교한다는 점이다.

 

저번 문제에서는, Comparator 을 implements 하는 CustomComparator 객체를 만들어서 해결했었는데, 이번엔 람다 식 연습도 할 겸, 람다식을 이용해서 풀어봤다.

 

람다식으로 풀어서 primitive 타입의 int 배열로 해결 할 수 있었다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    public static void main(String[] args) throws IOException {
        //2차원 평면 위의 점 N개가 주어지고 좌표를 y좌표가 증가하는 순으로, y좌표가 같다면 x좌표가 증가하는 순으로
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        int N = Integer.valueOf(br.readLine());
        int[][] coordinate = new int[N][2];

        for (int i = 0; i < N; i++) {
            String[] input = br.readLine().split(" ");
            coordinate[i][0] = Integer.valueOf(input[0]);
            coordinate[i][1] = Integer.valueOf(input[1]);
        }
  //      System.out.println(Arrays.deepToString(coordinate));
        Arrays.sort(coordinate, (coord1, coord2) -> {
            if (coord1[1] == coord2[1]) {
                return coord1[0] - coord2[0];
            } else {
                return coord1[1] - coord2[1];
            }
        });
  //      System.out.println(Arrays.deepToString(coordinate));

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


    }
}

 

 

Comparator 구현해서 해결

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
    public static void main(String[] args) throws IOException {
        //2차원 평면 위의 점 N개가 주어지고 좌표를 y좌표가 증가하는 순으로, y좌표가 같다면 x좌표가 증가하는 순으로
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        int N = Integer.valueOf(br.readLine());
        Integer[][] coordinate = new Integer[N][2];

        for (int i = 0; i < N; i++) {
            String[] input = br.readLine().split(" ");
            coordinate[i][0] = Integer.valueOf(input[0]);
            coordinate[i][1] = Integer.valueOf(input[1]);
        }
        //      System.out.println(Arrays.deepToString(coordinate));
        Arrays.sort(coordinate,new CustomComparator());
        //      System.out.println(Arrays.deepToString(coordinate));

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


    }

    static class CustomComparator implements Comparator<Integer[]> {
        @Override
        public int compare(Integer[] a1, Integer[] a2) {
            if (a1[1].equals(a2[1])) {
                return a1[0] - a2[0];
            } else {
                return a1[1] - a2[1];
            }
        }
    }

}