結果

問題 No.1103 Directed Length Sum
ユーザー tatt61880tatt61880
提出日時 2021-04-05 16:39:17
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 2,499 ms / 3,000 ms
コード長 1,433 bytes
コンパイル時間 2,715 ms
コンパイル使用メモリ 149,860 KB
実行使用メモリ 394,240 KB
最終ジャッジ日時 2024-09-16 12:25:50
合計ジャッジ時間 24,005 ms
ジャッジサーバーID
(参考情報)
judge6 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1,349 ms
394,240 KB
testcase_03 AC 1,030 ms
237,952 KB
testcase_04 AC 1,363 ms
135,680 KB
testcase_05 AC 2,499 ms
232,064 KB
testcase_06 AC 811 ms
89,472 KB
testcase_07 AC 137 ms
24,320 KB
testcase_08 AC 234 ms
35,200 KB
testcase_09 AC 77 ms
16,768 KB
testcase_10 AC 346 ms
46,720 KB
testcase_11 AC 1,436 ms
147,584 KB
testcase_12 AC 827 ms
90,368 KB
testcase_13 AC 488 ms
48,256 KB
testcase_14 AC 57 ms
13,568 KB
testcase_15 AC 598 ms
71,168 KB
testcase_16 AC 1,678 ms
165,376 KB
testcase_17 AC 1,785 ms
172,288 KB
testcase_18 AC 384 ms
46,592 KB
testcase_19 AC 1,544 ms
151,168 KB
testcase_20 AC 122 ms
19,456 KB
testcase_21 AC 215 ms
32,128 KB
testcase_22 AC 1,209 ms
124,160 KB
testcase_23 AC 691 ms
76,032 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

func main()
	const mod: int :: 1000000007
	class Node()
		+var parent: Node
		+var children: list<Node>
		+var depth: int
		+var num: int
		*func ctor()
			do me.parent :: null
			do me.children :: #list<Node>
		end func
		+func calcDepth()
			if(me.parent =& null)
				do me.depth :: 1
			else
				do me.depth :: me.parent.depth + 1
			end if
			do me.children.head()
			for i(0, ^me.children - 1)
				do me.children.get().calcDepth()
				do me.children.next()
			end for
		end func
		+func calcNum()
			if(^me.children = 0)
				do me.num :: 1
				ret
			end if
			do me.children.head()
			for i(0, ^me.children - 1)
				do me.children.get().calcNum()
				do me.num :+ me.children.get().num
				do me.children.next()
			end for
			do me.num :+ 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
	var as: []int :: #[n - 1]int
	var bs: []int :: #[n - 1]int
	for i(0, n - 2)
		var a: int :: cui@inputInt() - 1
		var b: int :: cui@inputInt() - 1
		do as[i] :: a
		do bs[i] :: b
		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]
		end if
	end for
	
	do root.calcDepth()
	do root.calcNum()
	
	var ans: int :: 0
	for i(0, n - 2)
		do ans :+ nodes[as[i]].depth * nodes[bs[i]].num
		do ans :% mod
	end for
	do cui@print("\{ans}\n")
end func
0