[백준] 2346번: 풍선 터트리기 - Kotlin[코틀린]

문제

 

2346번: 풍선 터뜨리기

 

---

풀이

 

 이 문제는 Deque(덱)을 이용해 풀이하면 된다. 덱은 양쪽에서 삽입과 삭제가 모두 가능한 자료구조이다.

 

 풍선의 번호와 풍선에 적혀 있는 숫자를 저장할 풍선 data class를 정의하고, 이를 이용하여 입력 받은 수를 덱에 저장한다. 그 후, 덱이 비어질 때까지 반복하면서 풍선을 터뜨리는 순서를 구하면 된다.

 

 덱의 첫번째 풍선을 터뜨리고, 풍선에 적힌 숫자에 따라 덱을 업데이트한다. 숫자가 양수인 경우, 덱의 앞부분에 있는 풍선을 'poll()' 하면서 그 풍선을 덱의 뒷부분에 'add()' 해주면 된다. 예를 들어, 덱에 있는 숫자가 [2, 3, -1, -2, 1]인 경우, 2만큼 이동시켜주면 [-1, -2, 1, 2, 3]이 된다. 음수인 경우에는 이와 반대로 처리해준다.

 

---

코드

 

import java.util.*

data class Balloon(val idx: Int, val move: Int)

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

    val num = br.readLine().toInt()
    val deque : Deque<Balloon> = ArrayDeque()
    val st = StringTokenizer(br.readLine())
    for(i in 1 .. num) {
        deque.add(Balloon(i, st.nextToken().toInt()))
    }

    val sb = StringBuilder()

    while (true) {
        val cur = deque.poll()
        sb.append("${cur.idx} ")
        val move = cur.move

        if(deque.isEmpty()) break

        if(move > 0) {
            for(i in 1 until move) {
                deque.addLast(deque.pollFirst())
            }
        } else {
            for(i in move until 0) {
                deque.addFirst(deque.pollLast())
            }
        }
    }

    bw.write(sb.toString())
    bw.flush()
    bw.close()
    br.close()
}