結果

問題 No.1637 Easy Tree Query
ユーザー tatt61880
提出日時 2023-09-17 12:38:43
言語 Kuin
(KuinC++ v.2021.9.17)
結果
AC  
実行時間 367 ms / 2,000 ms
コード長 835 bytes
コンパイル時間 6,639 ms
コンパイル使用メモリ 148,224 KB
実行使用メモリ 28,032 KB
最終ジャッジ日時 2024-07-04 08:20:55
合計ジャッジ時間 16,357 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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)
	
	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)
		var gCur: list<int> :: g[current]
		do gCur.head()
		do @cnts[current] :: 1
		for i(0, ^gCur - 1)
			var sub: int :: gCur.get()
			do gCur.next()
			if(@cnts[sub] = 0)
				do bfs(g, sub)
				do @cnts[current] :+ @cnts[sub]
			end if
		end for
	end func
end func
0