Submission #74240894

Source Code Expand

from sys import stdin
input_int = lambda: int(stdin.readline())
input_ints = lambda: list(map(int,stdin.readline().split()))

n = input_int()
a = input_ints()

ans = [0] * n
used = [False] * n
for i in range(n):
    if not used[i]:
        stack = [i]
        while not used[stack[-1]]:
            used[stack[-1]] = True
            stack.append(a[stack[-1]]-1)
        if ans[stack[-1]] == 0:
            val = len(stack) - 1 - stack.index(stack[-1])
            for x in stack:
                ans[x] = val
        else:
            val = ans[stack[-1]]
            for x in stack:
                ans[x] = val
print(*ans)

Submission Info

Submission Time
Task D - Telephone Game of Messages
User HumbleRobo
Language Python (PyPy 3.11-v7.3.20)
Score 366
Code Size 648 Byte
Status AC
Exec Time 212 ms
Memory 158060 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 366 / 366
Status
AC × 3
AC × 51
Set Name Test Cases
Sample sample01.txt, sample02.txt, sample03.txt
All sample01.txt, sample02.txt, sample03.txt, in01.txt, in02.txt, in03.txt, in04.txt, in05.txt, in06.txt, in07.txt, in08.txt, in09.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in48.txt
Case Name Status Exec Time Memory
in01.txt AC 50 ms 79284 KiB
in02.txt AC 49 ms 79268 KiB
in03.txt AC 48 ms 79340 KiB
in04.txt AC 48 ms 79548 KiB
in05.txt AC 48 ms 79276 KiB
in06.txt AC 49 ms 79380 KiB
in07.txt AC 203 ms 157716 KiB
in08.txt AC 191 ms 140700 KiB
in09.txt AC 50 ms 79144 KiB
in10.txt AC 210 ms 140324 KiB
in11.txt AC 206 ms 140620 KiB
in12.txt AC 205 ms 157632 KiB
in13.txt AC 193 ms 157772 KiB
in14.txt AC 212 ms 140480 KiB
in15.txt AC 183 ms 138696 KiB
in16.txt AC 195 ms 139836 KiB
in17.txt AC 50 ms 79308 KiB
in18.txt AC 192 ms 157920 KiB
in19.txt AC 184 ms 140480 KiB
in20.txt AC 195 ms 157788 KiB
in21.txt AC 187 ms 140240 KiB
in22.txt AC 50 ms 79304 KiB
in23.txt AC 52 ms 86216 KiB
in24.txt AC 48 ms 79544 KiB
in25.txt AC 56 ms 90832 KiB
in26.txt AC 47 ms 79400 KiB
in27.txt AC 47 ms 79000 KiB
in28.txt AC 185 ms 140624 KiB
in29.txt AC 197 ms 158060 KiB
in30.txt AC 192 ms 140896 KiB
in31.txt AC 50 ms 79500 KiB
in32.txt AC 49 ms 79000 KiB
in33.txt AC 47 ms 79000 KiB
in34.txt AC 183 ms 138664 KiB
in35.txt AC 204 ms 152468 KiB
in36.txt AC 194 ms 156456 KiB
in37.txt AC 185 ms 140676 KiB
in38.txt AC 160 ms 146076 KiB
in39.txt AC 192 ms 156460 KiB
in40.txt AC 201 ms 157872 KiB
in41.txt AC 51 ms 79368 KiB
in42.txt AC 49 ms 79500 KiB
in43.txt AC 48 ms 79356 KiB
in44.txt AC 47 ms 79500 KiB
in45.txt AC 47 ms 79108 KiB
in46.txt AC 47 ms 79492 KiB
in47.txt AC 47 ms 79532 KiB
in48.txt AC 208 ms 155304 KiB
sample01.txt AC 50 ms 79492 KiB
sample02.txt AC 48 ms 79500 KiB
sample03.txt AC 48 ms 79400 KiB