結果

問題 No.2095 High Rise
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-08-19 01:34:33
言語 Rust
(1.77.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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 1 ms
6,812 KB
testcase_03 AC 1 ms
6,816 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 1 ms
6,944 KB
testcase_06 AC 1 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 1 ms
6,940 KB
testcase_09 AC 1 ms
6,940 KB
testcase_10 AC 1 ms
6,944 KB
testcase_11 AC 1 ms
6,944 KB
testcase_12 AC 2 ms
6,944 KB
testcase_13 AC 1 ms
6,940 KB
testcase_14 AC 1 ms
6,940 KB
testcase_15 AC 1 ms
6,940 KB
testcase_16 AC 2 ms
6,940 KB
testcase_17 AC 12 ms
6,944 KB
testcase_18 AC 8 ms
6,940 KB
testcase_19 AC 2 ms
6,944 KB
testcase_20 AC 12 ms
7,040 KB
testcase_21 AC 35 ms
11,520 KB
testcase_22 AC 147 ms
42,908 KB
testcase_23 AC 115 ms
42,876 KB
testcase_24 AC 111 ms
42,876 KB
testcase_25 AC 112 ms
42,908 KB
testcase_26 AC 92 ms
42,876 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
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