結果

問題 No.697 池の数はいくつか
ユーザー masaki.arizuka
提出日時 2019-09-22 23:43:15
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 2,727 ms / 6,000 ms
コード長 3,881 bytes
コンパイル時間 13,381 ms
コンパイル使用メモリ 378,040 KB
実行使用メモリ 222,080 KB
最終ジャッジ日時 2024-11-08 08:24:33
合計ジャッジ時間 29,517 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#[allow(unused_macros)]
macro_rules! debug {
    ($($a:expr),*) => {
        #[cfg(debug_assertions)]
        writeln!(&mut std::io::stderr(), concat!("[DEBUG] ", $(stringify!($a), "={:?} "),*), $($a),*);
    }
}

#[allow(unused_imports)]
use std::cmp::{min, max};
#[allow(unused_imports)]
use std::io::{stdout, stdin, BufWriter, Write};

#[derive(Debug)]
pub struct UnionFind {
    par: Vec<usize>,
}
impl UnionFind {
    pub fn new(n: usize) -> UnionFind {
        let mut vec = vec![0; n];
        for i in 0..n {
            vec[i] = i;
        }
        UnionFind { par: vec }
    }
    pub fn find(&mut self, x: usize) -> usize {
        if x == self.par[x] {
            x
        } else {
            let par = self.par[x];
            let res = self.find(par);
            self.par[x] = res;
            res
        }
    }
    pub fn same(&mut self, a: usize, b: usize) -> bool {
        self.find(a) == self.find(b)
    }
    pub fn unite(&mut self, a: usize, b: usize) {
        let apar = self.find(a);
        let bpar = self.find(b);
        self.par[apar] = bpar;
    }
}

fn main() {
    let out = std::io::stdout();
    let mut out = BufWriter::new(out.lock());
    macro_rules! puts {
        ($($format:tt)*) => (write!(out,$($format)*).unwrap());
    }

    let stdin = std::io::stdin();
    let mut sc = Scanner { reader: stdin.lock() };

    let h:usize = sc.read();
    let w:usize = sc.read();

    let mut f = vec![];
    for _ in 0..h {
        let r:Vec<usize> = sc.read_vec(w);
        f.push(r);
    }

    use std::collections::VecDeque;
    let mut uf = UnionFind::new(h*w);

    let mut vis = vec![vec![false; w]; h];
    for x in 0..h {
        for y in 0..w {
            if f[x][y] == 0 || vis[x][y] {
                continue;
            }
            let mut q = VecDeque::new();
            q.push_back((x,y));

            while q.len() > 0 {
                let (x, y) = q.pop_front().unwrap();
                let x = x as i64;
                let y = y as i64;

                let ds = vec![
                    (x+1, y),
                    (x, y+1),
                    (x-1, y),
                    (x, y-1),
                ];
                for (nx, ny) in ds {
                    if nx < 0 || ny < 0 {
                        continue;
                    }
                    let x = x as usize;
                    let y = y as usize;
                    let nx = nx as usize;
                    let ny = ny as usize;

                    if nx >= h || ny >= w {
                        continue;
                    }

                    if vis[nx][ny] {
                        continue;
                    }
                    if f[nx][ny] == 0 {
                        continue;
                    }
                    vis[nx][ny] = true;
                    uf.unite(x*w+y, nx*w+ny);
                    q.push_back((nx, ny));
                }
            }
        }
    }

    let mut ans = vec![];
    for x in 0..h {
        for y in 0..w {
            if f[x][y] == 1 {
                ans.push(uf.find(x*w+y));
            }
        }
    }
    ans.sort();
    ans.dedup();
    puts!("{}\n", ans.len());
}

pub struct Scanner<R> {
    reader: R,
}
impl<R: std::io::Read> Scanner<R> {
    pub fn read<T: std::str::FromStr>(&mut self) -> T {
        use std::io::Read;
        let buf: String = self
            .reader
            .by_ref()
            .bytes()
            .map(|b| b.unwrap() as char)
            .skip_while(|c| c.is_whitespace())
            .take_while(|c| !c.is_whitespace())
            .collect();
        buf.parse::<T>().ok().expect("Parse error.")
    }
    pub fn read_vec<T: std::str::FromStr>(&mut self, n: usize) -> Vec<T> {
        (0..n).map(|_| self.read()).collect()
    }
    pub fn chars(&mut self) -> Vec<char> {
        self.read::<String>().chars().collect()
    }
}
0