[백준] 12789번: 도키도키 간식드리미 - Kotlin[코틀린]

문제

 

12789번: 도키도키 간식드리미

 

---

풀이

 

 대기열의 왼쪽에 있는 공간을 스택으로 생각하여 풀이하면 된다. 스택은 선입후출 구조이므로 왼쪽 공간에 먼저 들어간 번호는 나중에 나온다.

 

 현재 도시락을 받을 번호를 나타내는 변수 cur을 1로 초기화하고 선언한다. 입력받은 대기열을 앞에서부터 순회하면서 cur과 같은 번호를 만나면 도시락을 받은 것으로 처리하고, cur을 1 증가시켜서 다음으로 받을 사람을 나타낸다. 만약 cur과 다른 번호를 만난 경우에는 스택에 추가한다. 이 때, 현재 스택에서 가장 앞에 있는 번호가 cur인 경우에는 스택을 비워주면 된다.

 

 대기열이 끝나면 현재 스택에 남아있는 사람들을 처리한다. 만약 cur과 다른 번호를 만난 경우에는 도시락을 나눠줄수 없기 때문에 종료한다.

 

 스택에 아무것도 남아있지않다면 모든 사람이 도시락을 받을 수 있는 것이기 때문에 "Nice"를 출력하고 아니라면 "Sad"를 출력한다.

 

---

코드

 

import java.util.*

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()

    val num = br.readLine().toInt()
    val stack = Stack<Int>()

    var cur = 1
    br.readLine().split(' ').map { it.toInt() }.forEach {
        if(it == cur) cur++
        else {
            while (stack.isNotEmpty()) {
                if(stack.peek() == cur) {
                    stack.pop()
                    cur++
                }
                else break
            }
            stack.push(it)
        }
    }

    while (stack.isNotEmpty()) {
        if(stack.pop() == cur) cur++
        else break
    }

    bw.write(if(stack.isEmpty()) "Nice" else "Sad")
    bw.flush()
    bw.close()
    br.close()
}