여니의 취준 준비/코딩테스트 (Python)

[22858] 원상 복구 in python

여니's 2022. 2. 28. 12:05

 

temp[D[i]-1]=s[i] 를 이용하면

풀 수 있었던 문제! 

 

n, k = map(int, input().split())
s = list(map(int, input().split()))
d = list(map(int, input().split()))
for _ in range(k):
    temp = [0 for _ in range(n)]
    for i in range(n):
        temp[d[i]-1] = s[i]
    s = temp
print(*s)