[Python] 풍선 터뜨리기 - 백준

해당 글은 백준 2346번 문제 '풍선 터뜨리기'를 다룬다.

문제

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

2346번: 풍선 터뜨리기

 

설명

쉽게 풀 수 있는 방법으로 원형큐로 생각해서 rotate 시키면서 답을 구해나갈 수 있는 문제이다.

첫번째 인덱스 값을 지워나가면서 진행하기에 while 다음문에 터뜨릴 풍선을 첫번째 인덱스값으로 위치시켜야 한다.

그래서 rotate를 사용해서 큐 전체를 원형으로 돌린다.

이동할 값이 적혀있는 paper 값이 0보다 크면, 첫번째 인덱스 값을 지웠으므로 -1을 해줘야한다.

 

 

코드

from collections import deque

n = int(input())
queue = deque(enumerate(map(int,input().split())))

while queue:
    index, paper = queue.popleft()
    if paper > 0:
        queue.rotate(-(paper-1))
    if paper < 0:
        queue.rotate(-paper)
    print(index+1, end=' ')