寝癖頭の解法

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

アルゴ式(beta版): C++による「動的計画法 (DP):【補充】練習問題集」- Q. マス目の経路最適化 (2)

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

C++による「動的計画法 (DP):【補充】練習問題集」- Q. マス目の経路最適化 (2)


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

Q. マス目の経路最適化 (2)

algo-method.com

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

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