본문 바로가기
코딩테스트 준비/백준

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

by mwzz6 2024. 12. 6.

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

 

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


1.  아이디어

 

좌표를 배열로 저장한 후 람다식으로 정렬하는 방식으로 구현했다.


2. 문제풀이

 

sort 메서드의 파라미터로 람다식을 Comparator로 넣어줘서 객체 정렬을 수행하면 간단하게 해결할 수 있다.


3. 코드

 

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;

        int N = Integer.parseInt(br.readLine());

        int[][] posArr = new int[N][2];
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            posArr[i][0] = Integer.parseInt(st.nextToken());
            posArr[i][1] = Integer.parseInt(st.nextToken());
        }

        Arrays.sort(posArr, (o1, o2) -> {
            if (o1[1] == o2[1]) return Integer.compare(o1[0], o2[0]);
            return Integer.compare(o1[1], o2[1]);
        });

        for (int[] pos : posArr) {
            sb.append(pos[0]).append(" ").append(pos[1]).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

4. 후기