寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

アルゴ式(beta版): C++による「動的計画法 (DP):【補充】練習問題集」- Q. 連続ボーナス (3)

アルゴ式(beta版)の「動的計画法 (DP):【補充】練習問題集」からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。

C++による「動的計画法 (DP):【補充】練習問題集」- Q. 連続ボーナス (3)


僕が作成、提出したコードは、以下のとおりです。

Q. 連続ボーナス (3)

algo-method.com

/*
アルゴ式(beta版): C++による「動的計画法 (DP):【補充】練習問題集」- Q. 連続ボーナス (3)
Q. 連続ボーナス (3)
https://algo-method.com/tasks/1123
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
int main(void){
    ll m,n,a;
    cin>>m>>n>>a;
    vector<vector<ll>> p(m,vector<ll>(n));
    for(ll i=0;i<m;i++){
        for(ll j=0;j<n;j++){
            cin>>p[i][j];
        }
    }
    const ll inf=1<<30;
    vector<vector<ll>> dp(n,vector<ll>(m,inf));
    vector<vector<ll>> ml(n,vector<ll>(m,inf));
    vector<vector<ll>> mr(n,vector<ll>(m,inf));
    for(ll i=0;i<m;i++){
        dp[0][i]=p[i][0];
    }
    for(ll i=1;i<m;i++){
        ml[0][i]=min(ml[0][i-1],dp[0][i-1]);
    }
    for(ll i=m-2;i>=0;i--){
        mr[0][i]=min(mr[0][i+1],dp[0][i+1]);
    }
    for(ll i=1;i<n;i++){
        for(ll j=0;j<m;j++){
            ll k=p[j][i]+min({ml[i-1][j],dp[i-1][j]-a,mr[i-1][j]});
            dp[i][j]=k;
        }
        for(ll j=1;j<m;j++){
            ml[i][j]=min(ml[i][j-1],dp[i][j-1]);
        }
        for(ll j=m-2;j>=0;j--){
            mr[i][j]=min(mr[i][j+1],dp[i][j+1]);
        }
    }
    ll ans=inf;
    for(ll i=0;i<m;i++){
        ans=min(ans,dp[n-1][i]);
    }
    cout<<ans<<endl;
    return 0;
}

設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com