[백준] 22858번: 원상 복구 (small) (Python)

문제 링크

https://www.acmicpc.net/problem/22858

22858번: 원상 복구 (small)

 

풀이 코드

• P 카드를 D의 규칙에 따라 이동하는 문제
• 원래 카드인 P를 다시 찾는 것이므로 결과 카드인 S를 D의 규칙 반대로 이동시키기.
• P[Di] 카드를 i번째로 옮기기 -> S[i] 카드를 Di로 옮기기

from copy import deepcopy
n, k = map(int, input().split())

s = [0] + list(map(int, input().split()))
d = [0] + list(map(int, input().split()))

p = [0] * (n + 1)

for _ in range(k):
    for i in range(1, n + 1):
        p[d[i]] = s[i]
    s = deepcopy(p)

print(*p[1:])