[BOJ] 17827 - 달팽이 리스트 (Python)

 

 

아이디어

 

이동 후 도착할 인덱스를 계산하여 해당 값을 출력한다.

 

 

풀이

 

import sys

input = sys.stdin.readline

def solution():
    N, M, V = map(int, input().split())
    C = tuple(map(int, input().split()))
    res = []
    for _ in range(M):
        K = int(input())
        if K >= N:
            res.append(C[(K-(V-1))%(N-V+1)+(V-1)])
        else:
            res.append(C[K])
    print(*res, sep='\n')

solution()