SWEA: 1225 암호생성기

문제

SW Expert Academy

 

문제 풀이

 8개의 숫자가 주어질 때 1 - 5의 숫자를 반복하면서 숫자를 감소한 후에 마지막 숫자가 0이 될 때까지 반복하는 문제이다. 문제는 `deque`를 활용하면 쉽게 풀 수 있다. 반복문을 돌면서 감소시킬 수 있는 수(코드에서는 delta라고 하였다.)를 통해 맨 앞의 숫자를 감소시키고 맨 뒤로 보낸다. 이때 `list`의 `pop(0)`와 `append`를 사용하여도 되지만 `deque`를 사용하는 이유는 `pop(0)`의 경우 시간 복잡도가 O(N)이기 때문이다.

 

 이 문제에서는 `list`와 `deque` 어느 것을 사용하여도 상관 없지만, `pop(0)`가 많을 경우 시간이 오래 걸리게 된다.

 

 

코드

from collections import deque


for t in range(1, 11):
    input()
    array = deque(list(map(int, input().split())))
    delta = 1

    while True:
        cur_num = array.popleft()
        if cur_num - delta <= 0:
            array.append(0)
            break
        else:
            array.append(cur_num - delta)

        delta += 1

        if delta > 5:
            delta = 1

    print('#' + str(t), *array)