[SWEA] 1225. [S/W 문제해결 기본] 7일차 - 암호생성기

1225. [S/W 문제해결 기본] 7일차 - 암호생성기

문제 풀이

• 암호를 생성하기 위해 아래의 방식 순으로 진행하였다.

1. 젤 앞의 원소를 -i$(1 <= i <= 5)$ 를 해줌
2. 제일 앞 원소를 제일 뒤에 붙임(arr[-1])
3. 앞의 원소를 제거
4. 만약 -i한 제일 앞 원소(arr[-1])가 0보다 작거나 같게 된다면 0으로 바꿔주고 while문을 멈춘다.

import sys
sys.stdin = open("input.txt", "r")

for _ in range(1, 11):
    n = int(input())
    arr = list(map(int, input().split()))

    while True:
        a = True
        for k in range(1, 6):
            arr[0] -= k
            arr.append(arr[0])
            arr.remove(arr[0])

            if arr[-1] <= 0:
                a = False
                arr[-1] = 0
                break

        if not a:
            break

    arr = map(str, arr)
    print(f"#{n} {' '.join(arr)}")