Submission #24476637


Source Code Expand

import sys
from collections import defaultdict, deque

def main(f):
  N, M = list(map(int, f.readline().split()))
  P = [None] + list(map(int, f.readline().split()))
  link = [[] for _ in range(10**5+1)]
  for _ in range(M):
    x, y = list(map(int, f.readline().split()))
    link[x].append(y)
    link[y].append(x)

  flag = [0] * (N+1)
  q = deque()

  result = 0
  for n in range(1, N+1):
    if flag[n] == 1:
      continue

    group = set()
    ps = set()
    q.append(n)
    while q:
      x = q.pop()
      if flag[x] == 1:
        continue
      group.add(x)
      ps.add(P[x])
      flag[x] = 1
      for y in link[x]:
        q.append(y)

    result += len(group.intersection(ps))
#    for i in group:
#      if P[i] in group:
#        result += 1

  print(result)

main(sys.stdin)

Submission Info

Submission Time
Task D - Equals
User enakai
Language PyPy3 (7.3.0)
Score 400
Code Size 833 Byte
Status AC
Exec Time 325 ms
Memory 139652 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 400 / 400
Status
AC × 4
AC × 23
Set Name Test Cases
Sample 0_000.txt, 0_001.txt, 0_002.txt, 0_003.txt
All 0_000.txt, 0_001.txt, 0_002.txt, 0_003.txt, 1_004.txt, 1_005.txt, 1_006.txt, 1_007.txt, 1_008.txt, 1_009.txt, 1_010.txt, 1_011.txt, 1_012.txt, 1_013.txt, 1_014.txt, 1_015.txt, 1_016.txt, 1_017.txt, 1_018.txt, 1_019.txt, 1_020.txt, 1_021.txt, 1_022.txt
Case Name Status Exec Time Memory
0_000.txt AC 325 ms 84560 KiB
0_001.txt AC 62 ms 74972 KiB
0_002.txt AC 56 ms 74948 KiB
0_003.txt AC 63 ms 75392 KiB
1_004.txt AC 119 ms 85088 KiB
1_005.txt AC 178 ms 96528 KiB
1_006.txt AC 251 ms 139652 KiB
1_007.txt AC 63 ms 75160 KiB
1_008.txt AC 63 ms 75432 KiB
1_009.txt AC 64 ms 75168 KiB
1_010.txt AC 62 ms 75444 KiB
1_011.txt AC 63 ms 75236 KiB
1_012.txt AC 59 ms 75424 KiB
1_013.txt AC 78 ms 82748 KiB
1_014.txt AC 81 ms 83056 KiB
1_015.txt AC 72 ms 82008 KiB
1_016.txt AC 74 ms 82792 KiB
1_017.txt AC 81 ms 82532 KiB
1_018.txt AC 119 ms 83700 KiB
1_019.txt AC 116 ms 97868 KiB
1_020.txt AC 118 ms 97528 KiB
1_021.txt AC 130 ms 96628 KiB
1_022.txt AC 257 ms 127348 KiB