백준 2346 풍선 터뜨리기 파이썬

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

2346번: 풍선 터뜨리기

 

 

 

핵심

이 문제를 쉽게 풀 수 있는 핵심은 deque() 자료형의 rotate() 메서드다.

rotate(n)는 deque 의 내장함수로, n 이 양수면 우측으로, 음수면 좌측으로 회전한다고 볼 수 있다. 

 

코드

import sys
from collections import deque
input = sys.stdin.readline

n = int(input())
q = deque(enumerate(map(int, input().split())))
ans = []

while q:
    idx, paper = q.popleft()
    ans.append((idx + 1))
    if paper > 0:
        q.rotate(-(paper - 1))
    else:
        q.rotate(-paper)
print(' '.join(map(str, ans)))

"""
5
3 2 1 -3 -1
"""

 

풀이

문제에서 얻고자 하는 것은 종이의 번호가 아니라, 풍선의 위치 (인덱스) 다.

그렇기 때문에 풍선의 인덱스를 우선 ans 리스트에 저장할 수 있도록 입력을 받아야 한다. 

enumerate() 메서드를 사용하면, (index, value) 형태로 리스트를 생성할 수 있다.

q = deque(enumerate(map(int, input().split())))

 

이제 종이의 숫자가 양수 / 음수 냐에 따라 회전 방향을 분기하면 된다. 

기존의 풍선을 터뜨리고 paper 의 숫자만큼 움직여야 하기 때문에

paper 가 양수일 땐 (paper - 1) 처리 후에 인덱스를 이동,

paper 가 음수 일 땐, paper 만큼 좌측으로 회전시키면 된다.