アルゴ式(beta版)の「動的計画法」【補充】論理的思考力を鍛える練習問題集からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。
マス目の経路最適化
/* C++による「動的計画法」【補充】論理的思考力を鍛える練習問題集の解答例 マス目の経路最適化 https://algo-method.com/tasks/856 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; int main(void){ ll n; cin>>n; vector<vector<ll>> abc(3,vector<ll>(n)); for(ll i=0;i<3;i++){ for(ll j=0;j<n;j++){ cin>>abc[i][j]; } } const ll inf=1000000000000000000; vector<vector<ll>> dp(3,vector<ll>(n,inf)); for(ll i=0;i<3;i++){ dp[i][0]=0; } for(ll k=0;k<n-1;k++){ for(ll i=0;i<3;i++){ for(ll j=0;j<3;j++){ ll num=abs(abc[i][k]-abc[j][k+1]); dp[j][k+1]=min(dp[j][k+1],dp[i][k]+num); } } } cout<<min({dp[0][n-1],dp[1][n-1],dp[2][n-1]})<<endl; return 0; }
設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com