1149: 多重部分和问题
时间限制: 1 Sec 内存限制: 128 MB提交: 160 解决: 88
题目描述
有n中不同大小的数字ai,每种各mi个。判断是否可以从这些数字之中选出若干使他们的大小恰好为K.
限制条件
1<=n<=100
1<=ai,mi<=100000
1<=k<=100000
输入
多组数据。
第一行n。第二行不同的数字ai.第三行对应数字拥有的个数
输出
能挑选若干恰好和为K则输出“Yes”,反之输出“No”.
样例输入
1
3
3 5 8
3 2 2
17
样例输出
Yes
#include<iostream>
#include<algorithm>
using namespace std;
#include<string.h>
int a[105],b[105],dp[100000];
int main()
{
int n,ai,mi,k,flag=0;
cin>>n;
while(n--)
{
cin>>ai;
for(int i=1;i<=ai;i++)
{
cin>>a[i];
}
for(int i=1;i<=ai;i++)
{
cin>>b[i];
}
cin>>k;
dp[100000]={0};
for(int i=1;i<=ai;i++)
{
if(dp[k]==k)//剪枝(如果找到就不需要往后找了)
{ flag=1;
cout<<"Yes"<<endl;
break;
}
for(int gg=b[i];gg>=0;gg--)//循环次数
for(int j=k;j>=a[i];j--)//物品数量有限从后往前找
{
if(dp[j]<dp[j-a[i]]+a[i])//状态转移
dp[j]=dp[j-a[i]]+a[i];
}
}
if(!flag)
cout<<"No"<<endl;
}
return 0;
}
Fighting!