結果

問題 No.2563 色ごとのグループ
コンテスト
ユーザー ゼット
提出日時 2023-12-20 23:51:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 400 ms / 2,000 ms
コード長 909 bytes
コンパイル時間 282 ms
コンパイル使用メモリ 82,356 KB
実行使用メモリ 111,868 KB
最終ジャッジ日時 2024-09-27 10:17:32
合計ジャッジ時間 8,205 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

class unif:
  def __init__(self,n):
    self.pare=[-1]*n
    self.size=[1]*n
  def root(self,x):
    while self.pare[x]!=-1:
      x=self.pare[x]
    return x
  def unite(self,u,v):
    rootu=self.root(u)
    rootv=self.root(v)
    if rootu!=rootv:
      if self.size[rootu]>=self.size[rootv]:
        self.pare[rootv]=rootu
        self.size[rootu]+=self.size[rootv]
      else:
        self.pare[rootu]=rootv
        self.size[rootv]+=self.size[rootu]
  def same(self,s,t):
    return self.root(s)==self.root(t)
N,M=map(int,input().split())
C=list(map(int,input().split()))
R=[[] for i in range(N)]
Z=unif(N)
for i in range(N):
  R[C[i]-1].append(i)
G=[[] for i in range(N)]
for i in range(M):
  a,b=map(int,input().split())
  if C[a-1]==C[b-1]:
    Z.unite(a-1,b-1)
result=0
for i in range(N):
  if len(R[i])==0:
    continue
  A=set()
  for j in R[i]:
    A.add(Z.root(j))
  result+=len(A)-1
print(result)
0