Today I Learned

오늘은 백준 코틀린으로 코딩 테스트 문제를 풀었다.

CT_회의실 배정

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

문제

한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.

첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다.
첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.

풀이 과정

• 그리디 알고리즘을 이용해 풀었다.

1. 우선 입력을 mutableList<Pair>에 담아준다.
   meetings.add(start to end)처럼 key to value로 쓰면 Pair를 좀 더 간단하고 쉽게 담을 수 있다.

2. 리스트에 있는 회의 시간을 종료시간을 기준으로 오름차순 정렬한다.

3. 겹치지 않는 회의를 골라내기 위해 하나씩 순회하면서 현재 회의의 시작 시간이 이전 회의의 종료 시간보다 크거나 같은 경우에만 선택해서 회의 수 max 값을 찾는다.

Kotlin 코드

fun main() {
    val n = readln().toInt()
    val meetings = mutableListOf<Pair<Int, Int>>()

    repeat(n) {
        val (start, end) = readln().split(" ").map { it.toInt() }
        meetings.add(start to end)
    }

    meetings.sortWith(compareBy({ it.second }, { it.first }))

    var count = 0
    var endTime = 0

    for (meeting in meetings) {
        if (meeting.first >= endTime) {
            endTime = meeting.second
            count++
        }
    }

    println(count)
}