結果

問題 No.1390 Get together
ユーザー tatt61880
提出日時 2021-03-28 08:54:22
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 208 ms / 2,000 ms
コード長 1,172 bytes
コンパイル時間 2,339 ms
コンパイル使用メモリ 148,984 KB
実行使用メモリ 6,912 KB
最終ジャッジ日時 2024-09-16 12:12:52
合計ジャッジ時間 6,979 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

func main()
	var n: int :: cui@inputInt()
	var m: int :: cui@inputInt()
	var uni: @UnionFind :: (#@UnionFind).init(m + 1)
	
	var ans: int :: 0
	var boxs: []int :: #[n + 1]int
	for(1, n)
		var b: int :: cui@inputInt()
		var c: int :: cui@inputInt()
		if(boxs[c] = 0)
			do boxs[c] :: b
		else
			var b0: int :: boxs[c]
			if(!uni.same(b, b0))
				do ans :+ 1
				do uni.unite(b, b0)
			end if
		end if
	end for
	do cui@print("\{ans}\n")
end func

class UnionFind()
	var parents: []int
	+func init(n: int): UnionFind
		do me.parents :: [-1].repeat(n)
		ret me
	end func
	+func size(x: int): int
		ret - me.parents[me.root(x)]
	end func
	+func same(x: int, y: int): bool
		ret me.root(x) = me.root(y)
	end func
	+func unite(x: int, y: int): bool
		do x :: me.root(x)
		do y :: me.root(y)
		if(x = y)
			ret false
		end if
		if(me.size(x) < me.size(y))
			do me.parents[y] :+ me.parents[x]
			do me.parents[x] :: y
		else
			do me.parents[x] :+ me.parents[y]
			do me.parents[y] :: x
		end if
		ret true
	end func
	
	func root(x: int): int
		if(me.parents[x] < 0)
			ret x
		else
			do me.parents[x] :: me.root(me.parents[x])
			ret me.parents[x]
		end if
	end func
end class
0