寝癖頭の解法

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

アルゴ式(beta版): C++による「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」- Q. カラフルに選ぶ

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

C++による「動的計画法 (DP):【補充】部分問題に持たせる情報の工夫」- Q. カラフルに選ぶ


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

Q. カラフルに選ぶ

algo-method.com

/*
アルゴ式(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