結果

問題 No.1028 闇討ち
ユーザー phspls
提出日時 2023-01-03 21:25:26
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 69 ms / 2,000 ms
コード長 1,497 bytes
コンパイル時間 11,408 ms
コンパイル使用メモリ 377,516 KB
実行使用メモリ 33,536 KB
最終ジャッジ日時 2024-11-27 02:24:44
合計ジャッジ時間 14,139 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

fn main() {
    let mut n = String::new();
    std::io::stdin().read_line(&mut n).ok();
    let n: usize = n.trim().parse().unwrap();
    let grid = (0..n).map(|_| {
            let mut temp = String::new();
            std::io::stdin().read_line(&mut temp).ok();
            let temp: Vec<usize> = temp.trim().split_whitespace().map(|s| s.parse::<usize>().unwrap()-1).collect();
            temp
        })
        .collect::<Vec<_>>();

    let mut totals = vec![0usize; n];
    let mut ldists = vec![vec![0usize; n]; n];
    let mut rdists = vec![vec![0usize; n]; n];
    for i in 0..n {
        for j in 0..n {
            totals[grid[i][j]] += j;
            if i > j {
                ldists[grid[i][j]][i-j-1] += 1;
            }
            if i+j+1 < n {
                rdists[grid[i][j]][i+j+1] += 1;
            }
        }
    }
    let mut dists = vec![vec![0usize; n]; n];
    for i in 0..n {
        let mut incval = rdists[i][0];
        for j in 0..n-1 {
            let prev = incval;
            incval += rdists[i][j+1];
            rdists[i][j+1] += rdists[i][j] + prev;
        }
        let mut incval = ldists[i][n-1];
        for j in (1..n).rev() {
            let prev = incval;
            incval += ldists[i][j-1];
            ldists[i][j-1] += ldists[i][j] + prev;
        }
        for j in 0..n {
            dists[i][j] = totals[i] + ldists[i][j] + rdists[i][j];
        }
    }
    println!("{}", dists.iter().map(|v| v.iter().min().unwrap()).sum::<usize>());
}
0