結果

問題 No.1637 Easy Tree Query
ユーザー tatt61880
提出日時 2023-09-17 12:00:10
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 356 ms / 2,000 ms
コード長 857 bytes
コンパイル時間 6,531 ms
コンパイル使用メモリ 148,488 KB
実行使用メモリ 29,568 KB
最終ジャッジ日時 2024-07-04 07:51:34
合計ジャッジ時間 16,282 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

var cnts: []int

func main()
	var n: int :: cui@inputInt()
	var q: int :: cui@inputInt()
	
	var g: []list<int> :: #[n]list<int>
	for i(0, n - 1)
		do g[i] :: #list<int>
	end for
	
	for(1, n - 1)
		var a: int :: cui@inputInt() - 1
		var b: int :: cui@inputInt() - 1
		do g[a].add(b)
		do g[b].add(a)
	end for
	
	do @cnts :: #[n]int
	do bfs(g, 0, -1)
	
	var ans: int :: 0
	for(1, q)
		var p: int :: cui@inputInt() - 1
		var x: int :: cui@inputInt()
		do ans :+ @cnts[p] * x
		do cui@print("\{ans}\n")
	end for
	
	func bfs(g: []list<int>, current: int, parent: int): int
		var gc: list<int> :: g[current]
		do gc.head()
		var cnt: int :: 1
		for i(0, ^gc - 1)
			var neighbor: int :: gc.get()
			do gc.next()
			if(neighbor = parent)
				skip i
			end if
			do cnt :+ bfs(g, neighbor, current)
		end for
		do @cnts[current] :: cnt
		ret cnt
	end func
end func
0