문제
A사는 창립기념일 이벤트로 비밀번호 맞추기 대회를 열어, 최대 10개인 비밀번호를 맞추는 사람에게 기념품을 제공하기로 했다.
기념품을 받을 수 있도록 다음 조건에 맞는 비밀번호 찾기 프로그램을 작성하시오.
-
1000이하의 숫자 N개가 주어진다. 이때 시작 숫자가 정해지고, 첫 번째 지정 위치가 된다.
-
지정 위치부터 M번째 칸을 추가한다. 여기에 앞칸의 숫자와 뒤로 밀려난 칸의 숫자를 더해 넣는다. 추가된 칸이 새로운 지정 위치가 된다. 밀려난 칸이 없으면 시작 숫자와 더한다.
-
이 작업을 K회 반복하는데, M칸 전에 마지막 숫자에 이르면 남은 칸수는 시작 숫자부터 이어간다.
-
마지막 숫자부터 역순으로 숫자를 출력하면 비밀번호가 된다. 숫자가 10개 이상인 경우 10개까지만 출력한다.
다음은 N, M, K가 6, 3, 3이고, 주어진 숫자가 6, 2, 4, 9, 1, 5인 경우의 예이다. 6이 시작 숫자이자 첫번째 지정 위치가 된다.
[입력]
첫 줄에 테스트케이스의 수 T가 주어진다. 1<=T<=50
다음 줄부터 테스트 케이스의 별로 첫 줄에 N, M, K가, 다음 줄에 1000이하의 자연수 N개가 주어진다. 3<=N, M, K<=1000
[출력]
각 줄마다 "#T" (T는 테스트 케이스 번호)를 출력한 뒤, 답을 출력한다.
풀이
원소를 추가시킬 인덱스 : index+=M
추가 전 리스트 길이 : len(a)
원소 추가 방식
1. 원소 추가하려는 곳이 리스트의 제일 끝이면(index == len(a)) 추가전의 리스트 중
인덱스 0,-1인 두 원소를 더한값을 apppend해준다.
2. 그 외는 index를 len(a)로 나눈것의 나머지값을 index로 저장해주고 그곳에 앞 뒤의 값을 더해준것을 insert해준다.
그 후에 뒤에서부터 10개의 원소를 출력시킨다.
코드
T = int(input())
for test_case in range(1, T+1):
N, M, K = map(int,input().split())
a = list(map(int, input().split()))
index = 0 #시작인덱스
first_num=a[0]
for _ in range(K):
index+=M
if index == len(a):
a.append(first_num + a[-1])
else:
index %= len(a)
a.insert(index, (a[index - 1] + a[index]))
print(f'#{test_case}', end=' ')
print(*a[-1:-11:-1])