좌표 정렬하기 2
1 초 | 256 MB | 95319 | 62652 | 53525 | 67.144% |
문제
2차원 평면 위의 점 N개가 주어진다. 좌표를 y좌표가 증가하는 순으로, y좌표가 같으면 x좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.
입력
첫째 줄에 점의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 i번점의 위치 xi와 yi가 주어진다. (-100,000 ≤ xi, yi ≤ 100,000) 좌표는 항상 정수이고, 위치가 같은 두 점은 없다.
출력
첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.
예제 1
5
0 4
1 2
1 -1
2 2
3 3
예제 2
1 -1
1 2
2 2
3 3
0 4
간단한 문제이다.
Arrays.sort()와 Comparator를 이용하면 금방 풀 수 있다.
이차원 배열 arr[N][2]이 있다. (N은 주어진 좌표 개수, 2는 x,y 좌표를 의미)
배열 arr을 Arrays.sort()를 이용하여 정렬할 때, 우리만의 정렬 기준(Comparator)을 선언해 준다.
우선 arr[N][2]를 열어보면
아래와 같이 배열 안에 작은 배열이 있는 식으로 볼 수 있다.
arr = { {1, 2} ,
{2, -1},
{-1, 2},
.... }
Comparator에 우리가 선언할 compare은 arr 안의 작은 배열의 값을 비교하여 정렬하는 것이다.
// 핵심
Arrays.sort(arr, new Comparator<int[]>() {
@Override
public int compare(int[] a, int[] b){
if(a[1] == b[1]){ // arr[i][1] == arr[i+1][1] 두 점의 Y 좌표를 비교
return a[0]-b[0]; //같으면 X의 값을 비교하여 정렬
} else {
return a[1]-b[1]; //다르면 Y의 값을 비교하여 정렬
}
}
});
전체 코드는 다음과 같다.
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Comparator;
import java.util.Arrays;
public class BOJ11651_AlignTwoCoordinate {
public static void main(String[] args) throws Exception{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
int[][]arr = new int[n][2];
// 좌표를 배열에 담기
for(int i=0;i<n;i++){
String[] a = br.readLine().split(" ");
arr[i][0] = Integer.parseInt(a[0]);
arr[i][1] = Integer.parseInt(a[1]);
}
// 핵심
Arrays.sort(arr, new Comparator<int[]>() {
@Override
public int compare(int[] a, int[] b){
if(a[1] == b[1]){
return a[0]-b[0];
} else {
return a[1]-b[1];
}
}
});
for(int i=0; i<n; i++){
System.out.println(arr[i][0] + " " + arr[i][1]);
}
}
}
'공부 > 코딩 테스트' 카테고리의 다른 글
Java) Leetcode 17. Letter Combinations of a Phone Number (백트랙킹) (7) | 2025.04.11 |
---|---|
Java) Leetcode 5. Longest Palindromic Substring (DP 풀이) (1) | 2025.04.08 |
[Java] 백준 2751번 : 수 정렬하기 2 (0) | 2025.04.05 |
[Java] LeetCode 20. Valid Parentheses (0) | 2025.04.03 |
[Java] Long의 바이너리 값 팰린드롬 구하기 (0) | 2025.04.02 |