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

 

🔒 문제

 

 

🧠 생각의 흐름

 

입력값에 "위치가 같은 두 점은 없다" 라는 부분에서 중복 값 저장은 신경쓸 필요가 없다는 것을 알 수 있다.

 

총 2가지로 풀이할 수 있다.

(1) 일단 모든 숫자를 리스트나 배열에 담고 정렬하자

y 기준 정렬 -> x 기준 정렬 (stable) 

 

(2) SortedSet을 사용하자 ✔️

=> 애초에 담길 때 부터 우선순위에 의해 자동 정렬이 되는 자료구조로

정렬의 우선순위인 'comparator' 만 정의해주면 된다.

 

 

 

🔑 풀이

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter
import java.util.*
import kotlin.Comparator
import kotlin.collections.HashSet

enum class Position {
    xPos,
    yPos
}

fun main() {
    val bufferedReader = BufferedReader(InputStreamReader(System.`in`))
    val bufferedWriter = BufferedWriter(OutputStreamWriter(System.`out`))

    val testCaseCount : Int = bufferedReader.readLine().toInt()

    // 이 문제에서 set 을 사용할 수 있는 근거를 보여주는 문장은 "위치가 같은 두 점은 없다." 라고 생각합니다.
    // comparator 의 정의에 따라 같은 우선순위로 판단될 경우 중복 데이터를 insert 할 때 무시됩니다.

    val positionSet = HashSet<IntArray>(testCaseCount).toSortedSet(Comparator<IntArray> {firstPosition, secondPosition->
        when {
            // y 좌표를 비교해서 뒷놈이 더 작으면 바꿔요 (오름차순)
            firstPosition[Position.yPos.ordinal] > secondPosition[Position.yPos.ordinal] -> 1
            firstPosition[Position.yPos.ordinal] < secondPosition[Position.yPos.ordinal] -> -1

            else -> { // y 좌표가 같다면?
                    // x 좌표를 비교해서 정렬!
                // compareTo : Returns zero if this object is equal to the specified other object
                    firstPosition[Position.xPos.ordinal].compareTo(secondPosition[Position.xPos.ordinal])
            }
        }
    })

    for (i in 1..testCaseCount) {
         StringTokenizer(bufferedReader.readLine()).also {
            val x = it.nextToken().toInt()
            val y = it.nextToken().toInt()
            positionSet.add(intArrayOf(x, y))
        }
    }

    positionSet.forEach {
        bufferedWriter.write("${it[Position.xPos.ordinal]} ${it[Position.yPos.ordinal]}\n")
    }

    bufferedWriter.close()
}