白书-多重部分和问题

 

 

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;


const int max_n = 100 + 2;
const int max_a = 1e5 + 10;
const int max_m = 1e5 + 10;
const int max_K = 1e5 + 10;

int n,K;
int a[max_n],m[max_n];
int dp[max_K];

void solve()
{
    memset(dp,-1,sizeof(dp));
    // 需重复利用数组
    dp[0]=0;
    for(int i=1;i<=n;++i)
    {
        for(int j=0;j<=K;++j)
        {
            if(dp[j]>=0)
            {
                dp[j]=m[i];
            }
            else if(j<a[i] || dp[j-a[i]]<=0)
            {
                dp[j]=-1;
            }
            else
            {
                dp[j]=dp[ j-a[i] ] - 1;
            }
        }
    }

    if(dp[K]>=0)
    {
        printf("YES\n");
    }
    else
    {
        printf("NO\n");
    }
}

int main()
{
    scanf("%d %d",&n,&K);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&m[i]);
    }
    solve();
    return 0;
}


/*
3 17
3 5 8
3 2 2

*/