HHOJ 多重部分和问题(分组背包问题)

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！