Submission #15005813

---

Source Code Expand

use std::io::*;
use std::str::FromStr;

//https://qiita.com/tubo28/items/e6076e9040da57368845
fn read<T: FromStr>() -> T {
    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 dfs(n:usize,select:&mut Vec<Vec<bool>>,select_list:&mut Vec<bool>){

    if select_list.len() == n {
        select.push(select_list.to_vec());
        return;
    }

    for i in 0..2{
        if i == 0{
            select_list.push(true);
        }else{
            select_list.push(false);
        }
        dfs(n,select,select_list);
        select_list.pop();
    }
}

fn main() {
    let h:usize = read();
    let w:usize = read();
    let k:usize = read();
    let mut board:Vec<Vec<char>> = vec![vec![' '; w];h]; //列、行

    for i in 0..h{
        board[i] = read::<String>().chars().collect();
    }

    let mut h_select: Vec<Vec<bool>> = Vec::new();
    let mut select_list:Vec<bool> = Vec::new();
    dfs(h,&mut h_select,&mut select_list);
    let mut w_select: Vec<Vec<bool>> = Vec::new();
    let mut select_list:Vec<bool> = Vec::new();
    dfs(w,&mut w_select,&mut select_list);

    let mut ans = 0;

    for i in 0..h_select.len(){
        for j in 0..w_select.len(){
            let mut chk_board:Vec<Vec<char>> = board.clone(); //列、行
            for l in 0..h_select[i].len(){
                if h_select[i][l] == true {
                    for x in 0..w{
                        chk_board[l][x] = 'x';
                    }
                }
            }
            for m in 0..w_select[j].len(){
                if w_select[j][m] == true {
                    for x in 0..h{
                        chk_board[x][m] = 'x';
                    }
                }
            }
            //println!("h:{:?} j:{:?}", h_select[i], w_select[j]);
            //#の数を数える
            let mut black = 0;
            for x in 0..h{
                for y in 0..w{
                    if chk_board[x][y] == '#'{
                        black += 1;
                    }
                }
            }
            if black == k {
                ans += 1;
            }
            
        }
    }
    println!("{}",ans);
}

Submission Info

Judge Result