結果
問題 | No.1637 Easy Tree Query |
ユーザー |
|
提出日時 | 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 |
ソースコード
var cnts: []intfunc 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 forfor(1, n - 1)var a: int :: cui@inputInt() - 1var b: int :: cui@inputInt() - 1do g[a].add(b)do g[b].add(a)end for; 部分木のサイズdo @cnts :: #[n]intdo bfs(g, 0)var ans: int :: 0for(1, q)var p: int :: cui@inputInt() - 1var x: int :: cui@inputInt()do ans :+ @cnts[p] * xdo cui@print("\{ans}\n")end forfunc bfs(g: []list<int>, current: int)var gCur: list<int> :: g[current]do gCur.head()do @cnts[current] :: 1for 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 ifend forend funcend func