結果

問題 No.1390 Get together
ユーザー nessien
提出日時 2024-12-11 10:14:49
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 471 ms / 2,000 ms
コード長 1,218 bytes
コンパイル時間 408 ms
コンパイル使用メモリ 82,092 KB
実行使用メモリ 107,256 KB
最終ジャッジ日時 2024-12-11 10:15:00
合計ジャッジ時間 10,474 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff #

class unionfind:
    def __init__(self,n):
      self.n = n
      self.par = [-1]*n
      self.size = [1]*n

    def root(self,x):
         while True:
            if self.par[x] == -1:
                return x
            if self.par[self.par[x]] == -1:
                return self.par[x]
            self.par[x] = self.par[self.par[x]]
            x = self.par[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.par[rootu] = rootv
                self.size[rootv] += self.size[rootu]
            else:
                self.par[rootv] = rootu
                self.size[rootu] += self.size[rootv]

    def same(self,u,v):
        return self.root(u) == self.root(v)

    def siz(self, x):
        return self.size[self.root(x)]

N, M = map(int,input().split())
b = [None]*N 
c = [None]*N
P = [list() for i in range(N)]
for i in range(N):
	b[i], c[i] = map(int,input().split())
	b[i] -= 1 
	c[i] -= 1 
	P[c[i]].append(b[i])

uf = unionfind(M)
ans = 0
for i in range(N):
	if len(P[i]) > 1:
		for x in P[i]:
			if not uf.same(P[i][0],x):
				uf.unite(P[i][0],x)
				ans += 1
print(ans)
0