[swea] 1225. 암호생성기 / python 파이썬

🚩 큐(queue)

 

thinking

딱 봐도 큐라 큐로 풀었다. 

맨 앞의 원소 빼서 다시 뒤에 붙이면서 숫자 감소시키다 0보다 작아지면 반복문 빠져나오도록 조건 설정함

 

코드1 - 파이썬 리스트 방법 ( queue.pop(0) )

T = 10
for t in range(1, T+1):
    tc = int(input())
    queue = list(map(int, input().split()))

    i = 1
    while True:
        if i > 5:
            i = 1
        t = queue.pop(0) - i
        if t <= 0:
            queue.append(0)
            break
        queue.append(t)
        i += 1

    print("#{} {} {} {} {} {} {} {} {}".format(tc, *queue))

 

코드2 - deque 라이브러리 방법

import collections


T = 10
for t in range(1, T+1):
    tc = int(input())
    queue2 = collections.deque(list(map(int, input().split())), maxlen=8)

    i = 1
    while True:
        if i > 5:
            i = 1
        t = queue2.popleft() - i
        if t <= 0:
            queue2.append(0)
            break
        queue2.append(t)
        i += 1

    print("#{}".format(tc), end=" ")
    for q in queue2:
        print("{}".format(q), end=" ")
    print()
    # 출력형태는 여러가지로 할 수 있다.
    # print("#{}".format(tc), end=" ")
    # print(*queue2, end=" ")
    # print()