자바/백준

[백준 11650] 좌표 정렬하기

슈슈버거 2022. 11. 21. 16:39

문제

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

 

11650번: 좌표 정렬하기

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

www.acmicpc.net


내가 작성한 코드

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

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

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

        HashMap<Integer, ArrayList<Integer>> xy = new HashMap<>();

        for(int i = 0; i<num; i++){
            StringTokenizer st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken());
            int y = Integer.parseInt(st.nextToken());

            ArrayList<Integer> list = new ArrayList<>();

            if(xy.containsKey(x)) {
                list = xy.get(x);
            }
            list.add(y);

            xy.put(x, list);
        }

        Map<Integer, ArrayList<Integer>> sortedXY = new TreeMap<>(xy);
        ArrayList<Integer> x = new ArrayList<>(sortedXY.keySet());

        for(int i = 0; i<sortedXY.size(); i++){
            ArrayList<Integer> y = new ArrayList<>();

            y = sortedXY.get(x.get(i));
            y.sort(null);

            for(int j = 0; j<y.size(); j++){
                bw.write(x.get(i)+" "+y.get(j)+"\n");
            }
        }
        bw.close();
    }
}

 

오 HashMap에 value 칸에 여러 값을 어떻게 넣어야 할지 고민이었는데

ArrayList를 넣으면 되더라

신기하네