[Python] 백준 2346 - 풍선 터뜨리기 문제 풀이

Overview

BOJ 2346번 풍선 터뜨리기 Python 문제 풀이
분류: Data Structure (자료구조)

---

문제 페이지

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

---

풀이 코드

from sys import stdin
from collections import deque


def main():
    def input():
        return stdin.readline().rstrip()

    n = int(input())
    balloons = list(map(int, input().split()))

    dq = deque(list((i + 1, -balloons[i]) for i in range(n)))
    res = []
    while dq:
        idx, move = dq.popleft()
        if move < 0:
            move += 1
        res.append(idx)
        dq.rotate(move)

    print(*res, sep=' ')


if __name__ == "__main__":
    main()

deque의 rotate를 이용하여 풀이한다. 주의할 점은 풍선 안의 종이에 적힌 수 부호와 반대 방향으로 이동해야한다는 것과 오른쪽으로 이동할 때, 한 칸 적게 이동해야 한다는 것이다.