Submission #56974345

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sys
rints = lambda: list(map(int, sys.stdin.readline().split()))
rdecints = lambda: [e - 1 for e in rints()]
M = 60
N, K = rints()
X = rdecints()
A = rints()
dp = [[0] * N for _ in range(M)]
dp[0] = X
for i in range(M - 1):
    for j in range(N):
        dp[i + 1][j] = dp[i][dp[i][j]]
ind = list(range(N))
p = 0
while K > 0:
    if K & 1:
        for i in range(N):
            ind[i] = dp[p][ind[i]]
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

import sys
rints = lambda: list(map(int, sys.stdin.readline().split()))
rdecints = lambda: [e - 1 for e in rints()]

M = 60
N, K = rints()
X = rdecints()
A = rints()

dp = [[0] * N for _ in range(M)]
dp[0] = X
for i in range(M - 1):
    for j in range(N):
        dp[i + 1][j] = dp[i][dp[i][j]]

ind = list(range(N))
p = 0
while K > 0:
    if K & 1:
        for i in range(N):
            ind[i] = dp[p][ind[i]]
    p += 1
    K >>= 1

print(" ".join([str(A[ind[i]]) for i in range(N)]))

Submission Info

Judge Result