Kotlin 🍬 백준 13단계 :: 11650 번

좌표 정렬하기

문제   |

  2차원 평면 위의 점 N개가 주어진다. 좌표를 x좌표가 증가하는 순으로, x좌표가 같으면 y좌표가 증가하는 순서로 정렬한 다음 출력하는 프로그램을 작성하시오.

 

입력   |

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

 

출력   |

  첫째 줄부터 N개의 줄에 점을 정렬한 결과를 출력한다.

 

 

풀이  |

  sortedWith()와 인자로 compareBy()를 사용한다.

  예시 )

val sortedList = list.sortedWith(
    compareBy(
        { 조건 A }, // 1. 조건 A에 대한 정렬 수행 후
        { 조건 B }  // 2. 1에서 정렬된 리스트를 기반으로 조건 B에 대한 정렬 수행
    )
)

// list : 원본 리스트
// sortedList : 조건 A, B에 의해 정렬된 리스트

 

 

답안  |

import java.util.Scanner

fun main() = with(Scanner(System.`in`)) {
    val N = nextInt()
    val arrX = IntArray(N)
    val arrY = IntArray(N)
    
    for(i in 0 until N){
        val x = nextInt()
        val y = nextInt()
        arrX[i] = x
        arrY[i] = y
    }
    
    val arrXY = arrX.zip(arrY)
    val result = arrXY.sortedWith(compareBy({ it.first }, { it.second }))

    for (row in result) {
        print("${row.first} ${row.second}")
        println()
    }
}