和为K的组合(01背包)

给出N个正整数组成的数组A,求能否从中选出若干个,使他们的和为K。如果可以,输出:“Yes”,否则输出"No"。
Input
第1行:2个数N, K, N为数组的长度, K为需要判断的和(2 <= N <= 20,1 <= K <= 10^9)
第2 - N + 1行:每行1个数,对应数组的元素Aii (1 <= Aii <= 10^6)
Output
如果可以,输出:“Yes”,否则输出"No"。
Sample Input
5 13
2
4
6
8
10
Sample Output
No
以前也做过这种和为k的题目,以前记得用二分做的。然后这次也尝试用二分做,但是这一次不是确定的几个数字相加。这样二分就做不了了。然后想了想,这不就是一个简单的01背包嘛。k为背包容量,给出的数字是体积。就是看看最后可不可以吧背包填满就好了啊。模板题。
代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;

const int maxx=1e7+10;
ll a[100];
ll dp[maxx];
ll n,k;

int main()
{
	while(scanf("%lld%lld",&n,&k)!=EOF)
	{
		for(int i=0;i<n;i++)
		{
			scanf("%lld",&a[i]);
		}
		memset(dp,0,sizeof(dp));
		for(int i=0;i<n;i++)
		{
			for(int j=k;j>=a[i];j--)
			{
				dp[j]=max(dp[j],dp[j-a[i]]+a[i]);
			}
		}
		if(dp[k]==k) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
}

努力加油a啊,(o)/~

猜你喜欢

转载自blog.csdn.net/starlet_kiss/article/details/83054309