結果

問題 No.1103 Directed Length Sum
コンテスト
ユーザー tatt61880
提出日時 2021-04-06 00:46:29
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 2,083 ms / 3,000 ms
コード長 1,174 bytes
コンパイル時間 2,552 ms
コンパイル使用メモリ 149,388 KB
実行使用メモリ 316,160 KB
最終ジャッジ日時 2024-09-16 12:26:26
合計ジャッジ時間 21,029 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

func main()
	const mod: int :: 1000000007
	class Node()
		+var parent: Node
		+var children: list<Node>
		+var depth: int
		+var descendantsSize: int
		*func ctor()
			do me.parent :: null
			do me.children :: #list<Node>
		end func
		+func update()
			if(me.parent =& null)
				do me.depth :: 0
			else
				do me.depth :: me.parent.depth + 1
			end if
			do me.children.head()
			for i(0, ^me.children - 1)
				var child: Node :: me.children.get()
				do child.update()
				do me.descendantsSize :+ child.descendantsSize
				do me.children.next()
			end for
			do me.descendantsSize :+ 1
		end func
	end class
	
	var n: int :: cui@inputInt()
	var nodes: []Node :: #[n]Node
	for i(0, n - 1)
		do nodes[i] :: #Node
	end for
	for i(0, n - 2)
		var a: int :: cui@inputInt() - 1
		var b: int :: cui@inputInt() - 1
		do nodes[a].children.add(nodes[b])
		do nodes[b].parent :: nodes[a]
	end for
	var root: Node
	for i(0, n - 1)
		if(nodes[i].parent =& null)
			do root :: nodes[i]
			break i
		end if
	end for
	do root.update()
	
	var ans: int :: 0
	for i(0, n - 1)
		do ans :+ nodes[i].depth * nodes[i].descendantsSize
		do ans :% mod
	end for
	do cui@print("\{ans}\n")
end func
0