アルゴ式(beta版)の「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。
Q. カラフルに選ぶ
/* アルゴ式(beta版): C++による「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」- Q. カラフルに選ぶ Q. カラフルに選ぶ https://algo-method.com/tasks/1121 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; int main(void){ ll n,m; cin>>n>>m; vector<ll> w(n),c(n); for(ll i=0;i<n;i++){ cin>>w[i]; } for(ll i=0;i<n;i++){ cin>>c[i]; } vector<vector<ll>> v(256); for(ll i=0;i<n;i++){ v[c[i]].push_back(w[i]); } vector<vector<bool>> dp(257,vector<bool>(m+1,false)); dp[0][0]=true; for(ll i=0;i<256;i++){ dp[i+1]=dp[i]; for(auto j : v[i]){ for(ll k=0;k<=m;k++){ if(j+k<=m && dp[i][k]){ dp[i+1][j+k]=true; } } } } cout<<(dp[256][m] ? "Yes\n" : "No\n"); return 0; }
設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com