結果

問題 No.2095 High Rise
コンテスト
ユーザー LyricalMaestro
提出日時 2024-08-19 01:34:33
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 147 ms / 2,000 ms
コード長 2,003 bytes
コンパイル時間 10,873 ms
コンパイル使用メモリ 391,836 KB
実行使用メモリ 42,908 KB
最終ジャッジ日時 2024-08-19 01:34:51
合計ジャッジ時間 15,086 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused import: `collections::VecDeque`
 --> src/main.rs:5:11
  |
5 | use std::{collections::VecDeque, i64};
  |           ^^^^^^^^^^^^^^^^^^^^^
  |
  = note: `#[warn(unused_imports)]` on by default

ソースコード

diff # 

// https://atcoder.jp/contests/past202206-open/tasks/past202206_c

use proconio::input;

use std::{collections::VecDeque, i64};

fn main() {
    input! {
        n: usize,
        m: usize,
        a: [[i64; m]; n],
    }

    let mut dp: Vec<Vec<i64>> = vec![vec![i64::MAX;m]; n];
    for j in 0..m {
        dp[0][j] = 0;
    }

    //  各m部屋について1階からn階まで取り壊した時のAを計算(累積話)
    let mut cum_a_list: Vec<Vec<i64>> = vec![vec![i64::MAX;m]; n];
    for j in 0..m {
        let mut cum_a = 0;
        for i in 0..n {
            cum_a += a[i][j];
            cum_a_list[i][j] = cum_a;
        }
    }

    let mut min_cells = vec![vec![i64::MAX;m]; n];
    let mut min_cols = vec![i64::MAX;m];

    for i in 1..n {
        for j in 0..m {
            // dp[i][j]の値を計算する

            if m > 1 {
                //  途中の階からエレベーターを作る
                if i > 1 {
                    let x = min_cols[j] + cum_a_list[i][j];
                    dp[i][j] = dp[i][j].min(x);    
                }
            }

            //  1階からきちんとエレベータを作る
            dp[i][j] = dp[i][j].min(cum_a_list[i][j]);
        }

        // 次の階以降の計算をするための準備を実施
        let mut backwards = vec![0; m];
        let mut buck_min = i64::MAX;
        for j in (0..m).rev() {
            buck_min = buck_min.min(dp[i][j]);
            backwards[j] = buck_min;
        }

        let mut forward_min = i64::MAX;
        for j in 0..m {
            if j + 1 < m {
                min_cells[i][j] = forward_min.min(backwards[j + 1]);
            } else {
                min_cells[i][j] = forward_min;
            }
            forward_min = forward_min.min(dp[i][j])
        }

        for j in 0..m {
            min_cols[j] = min_cols[j].min(min_cells[i][j] - cum_a_list[i - 1][j]);
        }
    }

    let answer = dp[n - 1].iter().min().unwrap();
    println!("{}", answer);

}
0