結果

問題 No.901 K-ary εxtrεεmε
ユーザー ngtkanangtkana
提出日時 2024-05-07 18:58:03
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 215 ms / 3,000 ms
コード長 2,412 bytes
コンパイル時間 15,296 ms
コンパイル使用メモリ 378,796 KB
実行使用メモリ 19,200 KB
最終ジャッジ日時 2024-11-30 13:59:13
合計ジャッジ時間 21,324 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
19,200 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 117 ms
18,176 KB
testcase_08 AC 107 ms
18,176 KB
testcase_09 AC 103 ms
18,176 KB
testcase_10 AC 110 ms
18,176 KB
testcase_11 AC 114 ms
18,176 KB
testcase_12 AC 91 ms
18,304 KB
testcase_13 AC 96 ms
18,176 KB
testcase_14 AC 89 ms
18,176 KB
testcase_15 AC 90 ms
18,176 KB
testcase_16 AC 87 ms
18,176 KB
testcase_17 AC 164 ms
18,176 KB
testcase_18 AC 155 ms
18,176 KB
testcase_19 AC 160 ms
18,176 KB
testcase_20 AC 161 ms
18,176 KB
testcase_21 AC 163 ms
18,176 KB
testcase_22 AC 76 ms
18,176 KB
testcase_23 AC 77 ms
18,176 KB
testcase_24 AC 84 ms
18,176 KB
testcase_25 AC 79 ms
18,304 KB
testcase_26 AC 81 ms
18,176 KB
testcase_27 AC 213 ms
18,304 KB
testcase_28 AC 214 ms
18,304 KB
testcase_29 AC 215 ms
18,304 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

use proconio::input;
use std::cmp::Reverse;

fn main() {
    input! {
        n: usize,
        edges: [(usize, usize, i64); n - 1],
        q: usize,
    }
    let mut g = vec![Vec::new(); n];
    for &(i, j, _) in &edges {
        g[i].push(j);
        g[j].push(i);
    }
    let mut stack = vec![0];
    let mut parent = vec![usize::MAX; n];
    parent[0] = 0;
    let mut size = vec![1; n];
    while let Some(x) = stack.pop() {
        if x <= isize::MAX as usize {
            stack.push(!x);
            g[x].retain(|&y| y != parent[x]);
            for &y in &g[x] {
                if parent[y] != usize::MAX {
                    continue;
                }
                parent[y] = x;
                stack.push(y);
            }
        } else {
            let x = !x;
            g[x].sort_unstable_by_key(|&y| Reverse(size[y]));
            for &y in &g[x] {
                size[x] += size[y];
            }
        }
    }
    let mut sorted = Vec::new();
    let mut position = vec![usize::MAX; n];
    let mut head = (0..n).collect::<Vec<_>>();
    let mut stack = vec![0];
    while let Some(x) = stack.pop() {
        position[x] = sorted.len();
        sorted.push(x);
        if let Some(&y) = g[x].first() {
            head[y] = head[x];
        }
        stack.extend(g[x].iter().rev().copied());
    }
    let mut values = vec![0; n];
    for &(i, j, w) in &edges {
        let i = if parent[i] == j { i } else { j };
        values[position[i]] = w;
    }
    for i in (0..n - 1).rev() {
        values[i] += values[i + 1];
    }
    values.push(0);
    for _ in 0..q {
        input! {
            k: usize,
            mut vertices: [usize; k],
        }
        vertices.sort_unstable_by_key(|&x| position[x]);
        let mut ans = 0;
        for (mut i, mut j) in vertices
            .iter()
            .copied()
            .zip(vertices.iter().copied().cycle().skip(1))
        {
            loop {
                if position[i] > position[j] {
                    std::mem::swap(&mut i, &mut j);
                }
                if head[i] == head[j] {
                    ans += values[position[i] + 1] - values[position[j] + 1];
                    break;
                }
                ans += values[position[head[j]]] - values[position[j] + 1];
                j = parent[head[j]];
            }
        }
        ans /= 2;
        println!("{}", ans);
    }
}
0