結果

問題 No.1639 最小通信路
ユーザー StrorkisStrorkis
提出日時 2021-08-06 22:56:54
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 3,250 bytes
コンパイル時間 11,919 ms
コンパイル使用メモリ 402,340 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-17 03:08:34
合計ジャッジ時間 13,637 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

pub mod io {
    use std::io::{BufRead, ErrorKind};

    pub fn scan<R: BufRead>(r: &mut R) -> Vec<u8> {
        let mut res = Vec::new();
        loop {
            let buf = match r.fill_buf() {
                Ok(buf) => buf,
                Err(e) if e.kind() == ErrorKind::Interrupted => continue,
                Err(e) => panic!("{}", e), 
            };
            let (done, used, buf) = {
                match buf.iter().position(u8::is_ascii_whitespace) {
                    Some(i) => (i > 0 || res.len() > 0, i + 1, &buf[..i]),
                    None => (buf.is_empty(), buf.len(), buf),
                }
            };
            res.extend_from_slice(buf);
            r.consume(used);
            if done { return res; }
        }
    }

    #[macro_export]
    macro_rules! scan {
        ($r:expr, [$t:tt; $n:expr]) => {
            (0..$n).map(|_| scan!($r, $t)).collect::<Vec<_>>()
        };
        ($r:expr, [$t:tt]) => {
            scan!($r, [$t; scan!($r, usize)])
        };
        ($r:expr, ($($t:tt),*)) => {
            ($(scan!($r, $t)),*)
        };
        ($r:expr, Usize1) => {
            scan!($r, usize) - 1
        };
        ($r:expr, Bytes) => {
            io::scan($r)
        };
        ($r:expr, String) => {
            String::from_utf8(scan!($r, Bytes)).unwrap()
        };
        ($r:expr, $t:ty) => {
            scan!($r, String).parse::<$t>().unwrap()
        };
    }

    #[macro_export]
    macro_rules! input {
        ($($($v:ident)* : $t:tt),* $(,)?) => {
            let stdin = std::io::stdin();
            let ref mut reader = std::io::BufReader::new(stdin.lock());
            $(let $($v)* = scan!(reader, $t);)*
        };
    }
}

// reference: https://github.com/atcoder/ac-library
pub mod dsu {
    pub struct Dsu {
        n: usize,
        parent_or_size: Vec<usize>,
    }
    
    impl Dsu {
        pub fn new(n: usize) -> Dsu {
            Dsu { n, parent_or_size: vec![!1; n] }
        }
    
        pub fn merge(&mut self, a: usize, b: usize) -> usize {
            let (mut x, mut y) = (self.leader(a), self.leader(b));
            if x == y { return x; }
            if !self.parent_or_size[x] < !self.parent_or_size[y] {
                std::mem::swap(&mut x, &mut y);
            }
            self.parent_or_size[x] -= !self.parent_or_size[y];
            self.parent_or_size[y] = x;
            x
        }
    
        pub fn same(&mut self, a: usize, b: usize) -> bool {
            self.leader(a) == self.leader(b)
        }
    
        pub fn leader(&mut self, a: usize) -> usize {
            if self.parent_or_size[a] >= self.n { return a; }
            self.parent_or_size[a] = self.leader(self.parent_or_size[a]);
            self.parent_or_size[a]
        }
    
        pub fn size(&mut self, a: usize) -> usize {
            let x = self.leader(a);
            !self.parent_or_size[x]
        }
    }    
}

use dsu::Dsu;

fn main() {
    input! {
        n: usize,
        edges: [(Usize1, Usize1, String); n * (n - 1) / 2],
    }

    let mut ans = String::new();
    let mut dsu = Dsu::new(n);
    for (a, b, c) in edges {
        if !dsu.same(a, b) {
            dsu.merge(a, b);
            ans = c;
        }
    }
    println!("{}", ans);
}
0