[백준 - python] 2346번: 풍선 터뜨리기

문제 링크

2346번: 풍선 터뜨리기

문제 풀이

import sys
from collections import deque

input = sys.stdin.readline

n = int(input())
paper = deque(list(enumerate(map(int, input().split()))))
poped = []

while paper:
    index, num = paper.popleft()
    poped.append(index + 1)
    if num > 0:
        paper.rotate(-(num-1))
    else:
        paper.rotate(-num)
print(*poped)

2346 입출력 예시

*key point: deque 모듈의 rotate 메서드를 이용하면 쉽게 deque을 회전시킬 수 있다. 이때 회전 방향에 유의하여 (-)를 곱해주어야 하며, 종이에 쓰여있는 숫자가 양수인 경우에는 이미 popleft를 수행하면서 한번 회전 되었기 때문에 1을 빼주기도 해야한다. 또한, enumerate 함수를 활용하면 index를 함께 튜플로 저장하여 유용하게 활용할 수 있다. 아래의 코드는 enumerate 함수 없이 직접 index deque을 만들어 푼 경우이다.

import sys
from collections import deque

input = sys.stdin.readline

n = int(input())
paper = deque(list(map(int, input().split())))
index = deque([i for i in range(1, n + 1)])
poped = []

while paper:
    num = paper.popleft()
    poped.append(index.popleft())
    if num > 0:
        paper.rotate(-(num-1))
        index.rotate(-(num-1))
    else:
        paper.rotate(-num)
        index.rotate(-num)
print(*poped)