Submission #56968905

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()))
M = 60
N, K = rints()
A = [e - 1 for e in rints()]
dp = [[0] * N for _ in range(M + 1)]
dp[0] = A
for i in range(M):
    for j in range(N):
        dp[i + 1][j] = dp[i][dp[i][j]]
ans = 0
p = 0
while K > 0:
    if K & 1:
        ans = dp[p][ans]
    p += 1
    K >>= 1
print(ans + 1)
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

import sys
rints = lambda: list(map(int, sys.stdin.readline().split()))

M = 60
N, K = rints()
A = [e - 1 for e in rints()]

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

ans = 0
p = 0
while K > 0:
    if K & 1:
        ans = dp[p][ans]
    p += 1
    K >>= 1
print(ans + 1)

Submission Info

Judge Result