提出 #76968262

ソースコード 拡げる

class Union_find:
  def __init__(self, n):
    self.n = n
    self.parent_or_size = [-1]*n
  
  def root(self, a):
    if self.parent_or_size[a] < 0:
      return a
    self.parent_or_size[a] = self.root(self.parent_or_size[a])
    return self.parent_or_size[a]
  
  def merge(self, a, b):
    ra, rb = self.root(a), self.root(b)
    if ra == rb:
      return
    if -self.parent_or_size[ra] < -self.parent_or_size[rb]:
      ra, rb = rb, ra
    self.parent_or_size[ra] += self.parent_or_size[rb]
    self.parent_or_size[rb] = ra
  
  def issame(self, a, b):
    return self.root(a) == self.root(b)

N, M = map(int, input().split())
uf = Union_find(N)
for _ in range(M):
  u, v = map(lambda x: int(x)-1, input().split())
  uf.merge(u, v)

Q = int(input())
for _ in range(Q):
  s = int(input())-1
  print(-uf.parent_or_size[uf.root(s)])

提出情報

提出日時
問題 G - 友達の輪
ユーザ rk_haskey
言語 Python (PyPy 3.11-v7.3.20)
得点 366
コード長 867 Byte
結果 AC
実行時間 853 ms
メモリ 112236 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 366 / 366
結果
AC × 5
AC × 76
セット名 テストケース
Sample sample01.txt, sample02.txt, sample03.txt, sample04.txt, sample05.txt
All sample01.txt, sample02.txt, sample03.txt, sample04.txt, sample05.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, in49.txt, in50.txt, in51.txt, in52.txt, in53.txt, in54.txt, in55.txt, in56.txt, in57.txt, in58.txt, in59.txt, in60.txt, in61.txt, in62.txt, in63.txt, in64.txt, in65.txt, in66.txt, in67.txt, in68.txt, in69.txt, in70.txt, in71.txt
ケース名 結果 実行時間 メモリ
in01.txt AC 55 ms 79840 KiB
in02.txt AC 54 ms 79444 KiB
in03.txt AC 54 ms 79676 KiB
in04.txt AC 54 ms 79612 KiB
in05.txt AC 54 ms 79704 KiB
in06.txt AC 54 ms 79496 KiB
in07.txt AC 54 ms 79720 KiB
in08.txt AC 53 ms 79868 KiB
in09.txt AC 54 ms 79612 KiB
in10.txt AC 54 ms 79936 KiB
in11.txt AC 835 ms 111640 KiB
in12.txt AC 662 ms 111376 KiB
in13.txt AC 284 ms 110480 KiB
in14.txt AC 54 ms 81508 KiB
in15.txt AC 354 ms 110964 KiB
in16.txt AC 685 ms 110588 KiB
in17.txt AC 688 ms 110684 KiB
in18.txt AC 496 ms 110400 KiB
in19.txt AC 666 ms 110384 KiB
in20.txt AC 662 ms 110684 KiB
in21.txt AC 671 ms 110384 KiB
in22.txt AC 685 ms 110532 KiB
in23.txt AC 670 ms 110404 KiB
in24.txt AC 665 ms 110496 KiB
in25.txt AC 671 ms 110716 KiB
in26.txt AC 670 ms 110280 KiB
in27.txt AC 636 ms 109872 KiB
in28.txt AC 767 ms 111476 KiB
in29.txt AC 54 ms 79756 KiB
in30.txt AC 54 ms 81336 KiB
in31.txt AC 54 ms 80192 KiB
in32.txt AC 54 ms 80224 KiB
in33.txt AC 665 ms 110612 KiB
in34.txt AC 676 ms 110528 KiB
in35.txt AC 54 ms 79600 KiB
in36.txt AC 65 ms 90824 KiB
in37.txt AC 796 ms 111140 KiB
in38.txt AC 55 ms 79996 KiB
in39.txt AC 56 ms 81132 KiB
in40.txt AC 54 ms 79672 KiB
in41.txt AC 54 ms 79848 KiB
in42.txt AC 54 ms 79632 KiB
in43.txt AC 53 ms 79676 KiB
in44.txt AC 793 ms 111668 KiB
in45.txt AC 853 ms 112144 KiB
in46.txt AC 712 ms 110748 KiB
in47.txt AC 672 ms 110488 KiB
in48.txt AC 687 ms 110308 KiB
in49.txt AC 812 ms 111944 KiB
in50.txt AC 773 ms 111308 KiB
in51.txt AC 814 ms 112236 KiB
in52.txt AC 490 ms 110396 KiB
in53.txt AC 582 ms 110396 KiB
in54.txt AC 703 ms 110428 KiB
in55.txt AC 700 ms 110316 KiB
in56.txt AC 66 ms 92004 KiB
in57.txt AC 66 ms 91908 KiB
in58.txt AC 664 ms 110544 KiB
in59.txt AC 55 ms 81404 KiB
in60.txt AC 55 ms 81012 KiB
in61.txt AC 54 ms 79756 KiB
in62.txt AC 54 ms 79840 KiB
in63.txt AC 477 ms 110244 KiB
in64.txt AC 479 ms 109888 KiB
in65.txt AC 54 ms 79944 KiB
in66.txt AC 53 ms 79668 KiB
in67.txt AC 253 ms 110116 KiB
in68.txt AC 256 ms 110328 KiB
in69.txt AC 54 ms 79852 KiB
in70.txt AC 54 ms 79812 KiB
in71.txt AC 665 ms 110524 KiB
sample01.txt AC 54 ms 79616 KiB
sample02.txt AC 54 ms 79756 KiB
sample03.txt AC 54 ms 79672 KiB
sample04.txt AC 54 ms 79980 KiB
sample05.txt AC 54 ms 79496 KiB