結果
問題 | No.1637 Easy Tree Query |
ユーザー | akakimidori |
提出日時 | 2021-08-06 22:24:53 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 79 ms / 2,000 ms |
コード長 | 2,719 bytes |
コンパイル時間 | 11,205 ms |
コンパイル使用メモリ | 401,368 KB |
実行使用メモリ | 21,752 KB |
最終ジャッジ日時 | 2024-09-17 03:53:44 |
合計ジャッジ時間 | 14,596 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,376 KB |
testcase_02 | AC | 79 ms
14,336 KB |
testcase_03 | AC | 1 ms
5,376 KB |
testcase_04 | AC | 16 ms
7,168 KB |
testcase_05 | AC | 21 ms
8,320 KB |
testcase_06 | AC | 14 ms
6,400 KB |
testcase_07 | AC | 5 ms
5,376 KB |
testcase_08 | AC | 11 ms
6,016 KB |
testcase_09 | AC | 25 ms
10,880 KB |
testcase_10 | AC | 11 ms
5,376 KB |
testcase_11 | AC | 32 ms
12,416 KB |
testcase_12 | AC | 30 ms
10,880 KB |
testcase_13 | AC | 10 ms
5,376 KB |
testcase_14 | AC | 46 ms
10,112 KB |
testcase_15 | AC | 75 ms
12,544 KB |
testcase_16 | AC | 38 ms
12,544 KB |
testcase_17 | AC | 10 ms
5,376 KB |
testcase_18 | AC | 23 ms
8,064 KB |
testcase_19 | AC | 23 ms
8,704 KB |
testcase_20 | AC | 30 ms
10,880 KB |
testcase_21 | AC | 19 ms
8,320 KB |
testcase_22 | AC | 37 ms
14,080 KB |
testcase_23 | AC | 10 ms
5,376 KB |
testcase_24 | AC | 16 ms
7,808 KB |
testcase_25 | AC | 21 ms
8,064 KB |
testcase_26 | AC | 30 ms
11,392 KB |
testcase_27 | AC | 40 ms
14,080 KB |
testcase_28 | AC | 17 ms
6,912 KB |
testcase_29 | AC | 26 ms
9,856 KB |
testcase_30 | AC | 16 ms
7,808 KB |
testcase_31 | AC | 12 ms
5,376 KB |
testcase_32 | AC | 13 ms
6,400 KB |
testcase_33 | AC | 19 ms
6,784 KB |
testcase_34 | AC | 30 ms
21,752 KB |
ソースコード
// ---------- begin recurse ----------// reference// https://twitter.com/noshi91/status/1393952665566994434// https://twitter.com/shino16_cp/status/1393933468082397190pub fn recurse<A, R, F>(f: F) -> impl Fn(A) -> RwhereF: Fn(&dyn Fn(A) -> R, A) -> R,{fn call<A, R, F>(f: &F, a: A) -> RwhereF: Fn(&dyn Fn(A) -> R, A) -> R,{f(&|a| call(f, a), a)}move |a| call(&f, a)}// ---------- end recurse ----------// ---------- begin input macro ----------// reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8macro_rules! input {(source = $s:expr, $($r:tt)*) => {let mut iter = $s.split_whitespace();input_inner!{iter, $($r)*}};($($r:tt)*) => {let s = {use std::io::Read;let mut s = String::new();std::io::stdin().read_to_string(&mut s).unwrap();s};let mut iter = s.split_whitespace();input_inner!{iter, $($r)*}};}macro_rules! input_inner {($iter:expr) => {};($iter:expr, ) => {};($iter:expr, $var:ident : $t:tt $($r:tt)*) => {let $var = read_value!($iter, $t);input_inner!{$iter $($r)*}};}macro_rules! read_value {($iter:expr, ( $($t:tt),* )) => {( $(read_value!($iter, $t)),* )};($iter:expr, [ $t:tt ; $len:expr ]) => {(0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()};($iter:expr, chars) => {read_value!($iter, String).chars().collect::<Vec<char>>()};($iter:expr, bytes) => {read_value!($iter, String).bytes().collect::<Vec<u8>>()};($iter:expr, usize1) => {read_value!($iter, usize) - 1};($iter:expr, $t:ty) => {$iter.next().unwrap().parse::<$t>().expect("Parse error")};}// ---------- end input macro ----------use std::io::Write;fn run() {input! {n: usize,q: usize,e: [(usize1, usize1); n - 1],ask: [(usize1, u64); q],}let mut g = vec![vec![]; n];for (a, b) in e {g[a].push(b);g[b].push(a);}let dp = std::cell::RefCell::new(vec![0; n]);recurse(|f, (v, p, id): (usize, usize, u64)| -> u64 {let s = id;let mut id = id + 1;for &u in g[v].iter().filter(|u| **u != p) {id = f((u, v, id));}dp.borrow_mut()[v] = id - s;id})((0, n, 0));let dp = dp.into_inner();let out = std::io::stdout();let mut out = std::io::BufWriter::new(out.lock());let mut ans = 0;for (p, x) in ask {ans += dp[p] * x;writeln!(out, "{}", ans).ok();}}fn main() {run();}