結果

問題 No.556 仁義なきサルたち
ユーザー gyu-dongyu-don
提出日時 2017-08-19 08:20:47
言語 Rust
(1.77.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 2,546 bytes
コンパイル時間 1,260 ms
コンパイル使用メモリ 163,584 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-22 16:24:17
合計ジャッジ時間 2,170 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 0 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 1 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 7 ms
5,376 KB
testcase_14 AC 8 ms
5,376 KB
testcase_15 AC 8 ms
5,376 KB
testcase_16 AC 14 ms
5,376 KB
testcase_17 AC 15 ms
5,376 KB
testcase_18 AC 15 ms
5,376 KB
testcase_19 AC 14 ms
5,376 KB
testcase_20 AC 15 ms
5,376 KB
testcase_21 AC 14 ms
5,376 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unnecessary trailing semicolon
  --> main.rs:84:6
   |
84 |     };
   |      ^ help: remove this semicolon
   |
   = note: `#[warn(redundant_semicolons)]` on by default

warning: 1 warning emitted

ソースコード

diff #

#![allow(unused_macros, unused_imports, dead_code)]

use std::io::*;
use std::collections::*;

fn read_line() -> String {
    let mut s = String::new();
    stdin().read_line(&mut s).unwrap();
    s
}

macro_rules! from_iter {
    ($($a:ident),+ = $it:expr) => {
        $(let $a;)+
        {
            let mut _it = $it.into_iter();
            $($a = _it.next().unwrap();)+
                assert!(_it.next().is_none());
        }
    };
}

macro_rules! from_str {
    ($($a:ident : $t:ty),+ = $s:expr) => {
        $(let $a: $t;)+
        {
            let mut _it = $s.split_whitespace();
            $($a = _it.next().unwrap().parse().unwrap();)+
                assert!(_it.next().is_none());
        }
    };
}

macro_rules! from_line {
    ($($a:ident : $t:ty),+) => {
        $(let $a: $t;)+
        {
            let _line = read_line();
            let mut _it = _line.trim().split_whitespace();
            $($a = _it.next().unwrap().parse().unwrap();)+
                assert!(_it.next().is_none());
        }
    };
}

macro_rules! print_var_fmt {
    ($e:expr) => {
        concat!(stringify!($e), ":\t{}")
    };
    ($e:expr, $($es: expr),+) => {
        concat!(print_var_fmt!($e), ",\t", print_var_fmt!($($es),+))
    };
}

macro_rules! dbg {
    ($($es:expr),+) => {
        writeln!(&mut io::stderr(), print_var_fmt!($($es),*), $($es),*);
    };
}

fn main() {
    from_line!(n: usize, m: usize);

    let mut boss: Vec<u32> = (0..n as u32+1).collect();
    let mut piki: Vec<u32> = vec![1; n+1];

    fn root(boss: &mut Vec<u32>, i: usize) -> u32 {
        if boss[i] == i as u32 {
            return i as u32;
        } else {
            let j = boss[i] as usize;
            let t = root(boss, j);
            boss[i] = t as u32;
            return t as u32;
        }
    }

    fn unite(boss: &mut Vec<u32>, strong: usize, weak: usize) {
        let rw = root(boss, weak);
        let rs = root(boss, strong);
        if rw != rs {
            boss[rw as usize] = rs;
        }
    };

    for _ in 0..m {
        from_line!(a: usize, b: usize);
        let ta = root(&mut boss, a) as usize;
        let tb = root(&mut boss, b) as usize;
        if ta != tb {
            if piki[ta] > piki[tb] || piki[ta] == piki[tb] && ta < tb {
                piki[ta] += piki[tb];
                unite(&mut boss, ta, tb);
            } else {
                piki[tb] += piki[ta];
                unite(&mut boss, tb, ta);
            }
        }
    }
    for i in 1..n+1 {
        println!("{}", root(&mut boss, i));
    }
}
0