結果

問題 No.2563 色ごとのグループ
ユーザー timitimi
提出日時 2023-12-02 14:55:57
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 496 ms / 2,000 ms
コード長 772 bytes
コンパイル時間 478 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 136,056 KB
最終ジャッジ日時 2024-09-26 17:42:56
合計ジャッジ時間 7,816 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M=map(int, input().split())
C=list(map(int, input().split()))
par=[i for i in range(N)]
rank=[0]*(N)
friend=[0]*N
block=[0]*N
size=[1]*N
def find(x):
  if par[x]==x:
    return x
  else:
    par[x]=find(par[x])
    return par[x]
 
#同じ集合か判定
def same(x,y):
  return find(x)==find(y)
 
def union(x,y):
  x=find(x)
  y=find(y)
  if x==y:
    return 
  if rank[x]>rank[y]:
    par[y]=x
    size[x]+=size[y]
  else:
    par[x]=y
    size[y]+=size[x]
    if rank[x]==rank[y]:
      rank[y]+=1
 
 
for i in range(M):
  a,b=map(int,input().split())
  a-=1;b-=1
  if C[a]==C[b]:
    union(a,b)
    
ans=0
D={};E={}
for i in range(N):
  d=find(i)
  if d not in D:
    D[d]=1
    c=C[i]
    if c not in E:
      E[c]=0
    E[c]+=1

for e in E:
  ans+=E[e]-1
print(ans)
0