結果

問題 No.2095 High Rise
ユーザー tokumini_sstokumini_ss
提出日時 2022-10-07 21:51:17
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 420 ms / 2,000 ms
コード長 1,693 bytes
コンパイル時間 2,233 ms
コンパイル使用メモリ 206,348 KB
実行使用メモリ 19,072 KB
最終ジャッジ日時 2024-06-12 07:03:52
合計ジャッジ時間 6,240 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

int main() {
    int64_t N, M;
    cin >> N >> M;
    vector<vector<int64_t>> A(N, vector<int64_t>(M));
    for (int64_t i = 0; i < N; i++) {
        for (int64_t j = 0; j < M; j++) {
            cin >> A[i][j];
        }
    }

    // dp[i][j] := i回目まで行けるようにしたとき、直前ではj番目を取り壊したときの最小コスト
    vector<vector<int64_t>> dp(N + 1, vector<int64_t>(M, LLONG_MAX / 2));

    for (int64_t i = 0; i < N - 1; i++) {
        // ここでの最小コスト
        vector<int64_t> min_from_l(M, LLONG_MAX / 2);
        vector<int64_t> min_from_r(M, LLONG_MAX / 2);
        min_from_l[0] = dp[i][0];
        min_from_r[M - 1] = dp[i][M - 1];
        for (int64_t j = 1; j < M; j++) {
            min_from_l[j] = min(dp[i][j], min_from_l[j - 1]);
        }
        for (int64_t j = M - 2; j >= 0; j--) {
            min_from_r[j] = min(dp[i][j], min_from_r[j + 1]);
        }

        for (int64_t j = 0; j < M; j++) {
            // jを開けていて、そこから直接繋ぐ
            dp[i + 1][j] = min(dp[i + 1][j], dp[i][j] + A[i + 1][j]);

            // j以外を開けていて、そこから2段繋ぐ
            const int64_t min_l = (j == 0 ? LLONG_MAX / 2 : min_from_l[j - 1]);
            const int64_t min_r = (j == M - 1 ? LLONG_MAX / 2 : min_from_r[j + 1]);
            const int64_t other_min = (i == 0 ? 0 : min(min_l, min_r));
            dp[i + 1][j] = min(dp[i + 1][j], other_min + A[i][j] + A[i + 1][j]);
        }
    }

    if (N == 1) {
        cout << 0 << endl;
        return 0;
    }

    cout << *min_element(dp[N - 1].begin(), dp[N - 1].end()) << endl;
}
0