[SWEA] 5097 - 회전 (python)

문제

N개의 숫자로 이루어진 수열이 주어진다. 맨 앞의 숫자를 맨 뒤로 보내는 작업을 M번 했을 때, 수열의 맨 앞에 있는 숫자를 출력하는 프로그램을 만드시오.

[입력]

첫 줄에 테스트 케이스 개수 T가 주어진다. 1<=T<=50

다음 줄부터 테스트 케이스의 첫 줄에 N과 M이 주어지고, 다음 줄에 10억 이하의 자연수 N개가 주어진다. 3<=N<=20, N<=M<=1000,

[출력]

각 줄마다 "#T" (T는 테스트 케이스 번호)를 출력한 뒤, 번호를 출력한다.

풀이

arr 리스트안에 입력받은 숫자들을 저장하고 M번만큼 첫번째 원소를 꺼낸후 젤 뒤에 append시켜준다.

코드

T = int(input())
for test_case in range(1,T+1):
    N,M = map(int,input().split())
    arr = list(map(int,input().split()))
    for i in range(M):
        arr.append(arr.pop(0))
    print(f'#{test_case} {arr[0]}')