結果

問題 No.1054 Union add query
ユーザー uw_yu1rabbituw_yu1rabbit
提出日時 2021-03-13 02:35:30
言語 Rust
(1.77.0)
結果
AC  
実行時間 502 ms / 2,000 ms
コード長 1,564 bytes
コンパイル時間 1,039 ms
コンパイル使用メモリ 174,728 KB
実行使用メモリ 11,648 KB
最終ジャッジ日時 2024-04-22 16:11:31
合計ジャッジ時間 5,185 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 399 ms
5,376 KB
testcase_04 AC 443 ms
11,648 KB
testcase_05 AC 368 ms
5,376 KB
testcase_06 AC 348 ms
5,632 KB
testcase_07 AC 307 ms
5,632 KB
testcase_08 AC 244 ms
5,632 KB
testcase_09 AC 502 ms
11,648 KB
testcase_10 AC 108 ms
7,620 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable does not need to be mutable
  --> main.rs:31:13
   |
31 |         let mut b:i64 = read();
   |             ----^
   |             |
   |             help: remove this `mut`
   |
   = note: `#[warn(unused_mut)]` on by default

warning: 1 warning emitted

ソースコード

diff # 

#[allow(dead_code)]
#[allow(unused_imports)]
fn read<T: std::str::FromStr>() -> T {
    use std::io::*;
    let stdin = stdin();
    let stdin = stdin.lock();
    let token: String = stdin
        .bytes()
        .map(|c| c.expect("failed to read char") as char)
        .skip_while(|c| c.is_whitespace())
        .take_while(|c| !c.is_whitespace())
        .collect();
    token.parse().ok().expect("failed to parse token")
}

fn root (par:&Vec<usize>, mut x:usize) -> usize {
    while par[x] != x {
        x = par[x];
    }
    x
}
fn main(){
    let n:usize = read();
    let query:usize = read();
    let mut par:Vec<usize> = (0..n).collect();
    let mut sum = vec![0i64;n];
    let mut sz = vec![1;n];
    for _ in 0..query {
        let t:usize = read();
        let mut a:usize = read();
        let mut b:i64 = read();
        if t == 1 {
            a -= 1;
            let b = b as usize - 1;
            a = root(&par,a);
            let mut b = root(&par,b);
            if a != b {
                if sz[a] < sz[b] {
                    std::mem::swap(&mut a,&mut b);
                }
                sz[a] += sz[b];
                sum[b] -= sum[a];
                par[b] = a;
            }
        }else if t == 2 {
            a -= 1;
            a = root(&par,a);
            sum[a] += b;
        }else {
            a -= 1;
            let mut x = a;
            let mut ans = sum[x];
            while x != par[x] {
                x = par[x];
                ans += sum[x];
            }
            println!("{}",ans);
        }
    }
}
0