Submission #56970344

Source Code Expand

import sys

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

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

T = [0]
ind = {0: 0}
cycle_start = -1
for i in range(N):
    nxt = A[T[-1]]
    if nxt in ind:
        cycle_start = ind[nxt]
        break
    T.append(nxt)
    ind[nxt] = i + 1

if len(T) > K:
    print(T[K] + 1)
else:
    print(T[cycle_start + (K - cycle_start) % len(T[cycle_start:])] + 1)

Submission Info

Submission Time
Task D - Teleporter
User gigolo
Language Python (PyPy 3.10-v7.3.12)
Score 400
Code Size 435 Byte
Status AC
Exec Time 116 ms
Memory 157716 KiB

Judge Result

Set Name Sample Subtask1
Score / Max Score 0 / 0 400 / 400
Status
AC × 2
AC × 57
Set Name Test Cases
Sample sample_01.txt, sample_02.txt
Subtask1 sample_01.txt, sample_02.txt, sub1_01.txt, sub1_02.txt, sub1_03.txt, sub1_04.txt, sub1_05.txt, sub1_06.txt, sub1_07.txt, sub1_08.txt, sub1_09.txt, sub1_10.txt, sub1_11.txt, sub1_12.txt, sub1_13.txt, sub1_14.txt, sub1_15.txt, sub1_16.txt, sub1_17.txt, sub1_18.txt, sub1_19.txt, sub1_20.txt, sub1_21.txt, sub1_22.txt, sub1_23.txt, sub1_24.txt, sub1_25.txt, sub1_26.txt, sub1_27.txt, sub1_28.txt, sub1_29.txt, sub1_30.txt, sub1_31.txt, sub1_32.txt, sub1_33.txt, sub1_34.txt, sub1_35.txt, sub1_36.txt, sub1_37.txt, sub1_38.txt, sub1_39.txt, sub1_40.txt, sub1_41.txt, sub1_42.txt, sub1_43.txt, sub1_44.txt, sub1_45.txt, sub1_46.txt, sub1_47.txt, sub1_48.txt, sub1_49.txt, sub1_50.txt, sub1_51.txt, sub1_52.txt, sub1_53.txt, sub1_54.txt, sub1_55.txt
Case Name Status Exec Time Memory
sample_01.txt AC 58 ms 76040 KiB
sample_02.txt AC 58 ms 76484 KiB
sub1_01.txt AC 64 ms 82604 KiB
sub1_02.txt AC 75 ms 101524 KiB
sub1_03.txt AC 60 ms 81440 KiB
sub1_04.txt AC 84 ms 111576 KiB
sub1_05.txt AC 84 ms 111736 KiB
sub1_06.txt AC 75 ms 101316 KiB
sub1_07.txt AC 73 ms 99048 KiB
sub1_08.txt AC 82 ms 111840 KiB
sub1_09.txt AC 91 ms 120172 KiB
sub1_10.txt AC 83 ms 108436 KiB
sub1_11.txt AC 71 ms 93444 KiB
sub1_12.txt AC 87 ms 113820 KiB
sub1_13.txt AC 89 ms 115672 KiB
sub1_14.txt AC 111 ms 146528 KiB
sub1_15.txt AC 76 ms 101620 KiB
sub1_16.txt AC 83 ms 111652 KiB
sub1_17.txt AC 75 ms 101348 KiB
sub1_18.txt AC 85 ms 111760 KiB
sub1_19.txt AC 67 ms 86784 KiB
sub1_20.txt AC 110 ms 146124 KiB
sub1_21.txt AC 92 ms 120524 KiB
sub1_22.txt AC 85 ms 110768 KiB
sub1_23.txt AC 82 ms 109600 KiB
sub1_24.txt AC 102 ms 136268 KiB
sub1_25.txt AC 101 ms 136296 KiB
sub1_26.txt AC 84 ms 110740 KiB
sub1_27.txt AC 81 ms 106856 KiB
sub1_28.txt AC 89 ms 117160 KiB
sub1_29.txt AC 70 ms 90080 KiB
sub1_30.txt AC 81 ms 106988 KiB
sub1_31.txt AC 63 ms 81468 KiB
sub1_32.txt AC 76 ms 101464 KiB
sub1_33.txt AC 68 ms 88764 KiB
sub1_34.txt AC 86 ms 113464 KiB
sub1_35.txt AC 63 ms 81272 KiB
sub1_36.txt AC 96 ms 126052 KiB
sub1_37.txt AC 87 ms 114124 KiB
sub1_38.txt AC 106 ms 142120 KiB
sub1_39.txt AC 86 ms 113504 KiB
sub1_40.txt AC 66 ms 85228 KiB
sub1_41.txt AC 115 ms 152784 KiB
sub1_42.txt AC 111 ms 146304 KiB
sub1_43.txt AC 111 ms 146340 KiB
sub1_44.txt AC 106 ms 139328 KiB
sub1_45.txt AC 109 ms 143268 KiB
sub1_46.txt AC 85 ms 112292 KiB
sub1_47.txt AC 91 ms 118676 KiB
sub1_48.txt AC 110 ms 147768 KiB
sub1_49.txt AC 92 ms 121952 KiB
sub1_50.txt AC 85 ms 112512 KiB
sub1_51.txt AC 85 ms 110636 KiB
sub1_52.txt AC 73 ms 97320 KiB
sub1_53.txt AC 70 ms 93260 KiB
sub1_54.txt AC 116 ms 157716 KiB
sub1_55.txt AC 102 ms 135960 KiB