結果

問題 No.1345 Beautiful BINGO
ユーザー BrightLight
提出日時 2021-01-16 14:55:18
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 2,605 bytes
コンパイル時間 11,643 ms
コンパイル使用メモリ 379,560 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-27 17:11:46
合計ジャッジ時間 13,278 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 39 WA * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

fn read<T: std::str::FromStr>() -> T {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().parse().ok().unwrap()
}

fn read_vec<T: std::str::FromStr>() -> Vec<T> {
    read::<String>().split_whitespace()
        .map(|e| e.parse().ok().unwrap()).collect()
}

fn read_vec2<T: std::str::FromStr>(n: usize) -> Vec<Vec<T>> {
    (0..n).map(|_| read_vec()).collect()
}


fn main() {
    let nm: Vec<usize> = read_vec();
    let n = nm[0];
    let m = nm[1];
    let mut a: Vec<Vec<usize>> = read_vec2(n);

    let mut ans = 0;
    for _i in 0..m {
        let mut smallest = 9999999;
        let mut pos: (usize, usize, usize) = (999, 999, 999);
        for j in 0..n {
            let mut x = 0;
            let mut y = 0;    
            for k in 0..n {
                x += a[j][k];
                y += a[k][j];
            }
            if smallest > x && x != 0 {
                smallest = x;
                pos = (j, 999, 999);
            }
            if smallest > y && y != 0 {
                smallest = y;
                pos = (999, j, 999);
            }
        }
        let mut z1 = 0;
        let mut z2 = 0;
        for k in 0..n {
            z1 += a[k][k];
            z2 += a[k][n - k - 1];
        }
        if smallest > z1 && z1 != 0 {
            smallest = z1;
            pos = (999, 999, 1);
        }
        if smallest > z2 && z2 != 0 {
            smallest = z2;
            pos = (999, 999, 2);
        }

        ans += smallest;
        if pos.0 != 999 {
            for k in 0..n {
                a[pos.0][k] = 0;
            }
        } else if pos.1 != 999 {
            for k in 0..n {
                a[k][pos.1] = 0;
            }
        } else if pos.2 == 1 {
            for k in 0..n {
                a[k][k] = 0;
            }
        } else {
            for k in 0..n {
                a[k][n - k - 1] = 0;
            }
        }

        let mut zero = 0;
        for j in 0..n {
            let mut x = 0;
            let mut y = 0;    
            for k in 0..n {
                x += a[j][k];
                y += a[k][j];
            }
            if x == 0 {
                zero += 1;
            }
            if y == 0 {
                zero += 1;
            }
        }
        let mut z1 = 0;
        let mut z2 = 0;
        for k in 0..n {
            z1 += a[k][k];
            z2 += a[k][n - k - 1];
        }
        if z1 == 0 {
            zero += 1;
        }
        if z2 == 0 {
            zero += 1;
        }
        if zero >= m {
            break;
        }
    }
    println!("{}", ans);
}
0