結果

問題 No.977 アリス仕掛けの摩天楼
ユーザー tatt61880
提出日時 2021-02-25 00:32:07
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 162 ms / 2,000 ms
コード長 1,327 bytes
コンパイル時間 2,370 ms
コンパイル使用メモリ 148,968 KB
実行使用メモリ 14,208 KB
最終ジャッジ日時 2024-09-16 11:39:10
合計ジャッジ時間 4,209 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

func main()
	var n: int :: cui@inputInt()
	var uni: UnionFind :: (#UnionFind).init(n)
	
	var a: [][]int :: #[n, 0]int
	for(1, n - 1)
		var u: int :: cui@inputInt()
		var v: int :: cui@inputInt()
		do a[u] :~ [v]
		do a[v] :~ [u]
		do uni.unite(u, v)
	end for
	
	var ans: bool :: true
	if(uni.size(0) <> n)
		for i(0, n - 1)
			switch(^a[i])
			case 0
			case 2
				if(uni.size(i) <> n - 1)
					do ans :: false
					break i
				end if
			default
				do ans :: false
				break i
			end switch
		end for
	end if
	
	do cui@print(ans ?("Bob\n", "Alice\n"))
	
	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
end func
0