[SWEA] 1225 - 암호생성기 / Python / D3

KimYoungWoong·2022년 11월 2일
0
post-thumbnail

🚩문제 주소


📄풀이


한 사이클을 돌리는데 리스트 슬라이스를 사용하여 맨 앞 숫자를 떼어서 뒤에 붙여줍니다.
맨 앞 숫자가 0 이라면 check를 False로 바꿔줘서 while문을 종료시킨 후 답을 출력합니다.



👨‍💻코드


for tc in range(1, 11):
  n = int(input())
  data = list(map(int, input().split()))
  check = True

  while check:
    for i in range(1, 6):
      if data[0]-i > 0:
        data[0] = data[0]-i
        data = data[1:]+data[:1]
      else:
        data[0] = 0
        data = data[1:]+data[:1]
        check = False
        break
  
  print('#{} {} {} {} {} {} {} {} {}'.format(tc, *data))

profile
블로그 이전했습니다!! https://highero.tistory.com

0개의 댓글