자바/백준
[백준 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를 넣으면 되더라
신기하네