提出 #39377821

---

ソースコード 拡げる

#![allow(unused_imports)]

fn main() {
    input! {
        (h,w): (usize, usize),
        c: [Chars; h],
    }
    let mut d = UnionFind::new(h*w);
    for i in 0..h {
        for j in 0..w {
            if c[i][j] != 'o' { continue; }
            for (di, dj) in [(0,1), (1,0), (1,1)].iter() {
                let (ni, nj) = (i+di, j+dj);
                if ni == h || nj == w || c[ni][nj] != 'o' { continue; }
                d.union(i*w+j, ni*w+nj);
            }
            if i > 0 {
                let (ni, nj) = (i-1, j+1);
                if nj == w || c[ni][nj] != 'o' { continue; }
                d.union(i*w+j, ni*w+nj);
            }
        }
    }
    let mut tmp = vec![0; h*w];
    for l in d.into_labeling() {
        tmp[l] += 1;
    }
    let mut cnt = vec![0; h*w+1];
    for t in tmp {
        cnt[t] += 1;
    }
    let mut ans = (0, 0, 0);
    for i in 1..1000 {
        if let Some(v) = cnt.get(12*i*i) { ans.0 += v; }
        if let Some(v) = cnt.get(16*i*i) { ans.1 += v; }
        if let Some(v) = cnt.get(11*i*i) { ans.2 += v; }
    }
    println!("{} {} {}", ans.0, ans.1, ans.2);
}

use proconio::{
    input,
    marker::*,
};
use petgraph::unionfind::UnionFind;
use itertools::Itertools;

提出情報

ジャッジ結果