T^T找数字(搜索+二进制枚举)

题目链接:传送门
 
解题思路:看一眼n的大小，发现只有20，所以我们可以用决策搜索，或者二进制枚举解决，用搜索做法则需要判断第i个位置是否选取，然后当前的总和是多少，其实用二进制枚举也是这样做的，我们判断当前这个数是否选取，然后每次选取从1到n判断是否选取完成后，把选取的和与k进行比较，如果相等则直接按照给定选取的序列的顺序输出就行。其实两种做法的原理，或者说思路都是一样的。时间复杂度均为\(O(2^N)\)
搜索Code://8MS

#include<bits/stdc++.h>
using namespace std;
int n,k,a[25],vis[25];
bool fg;
void dfs(int loc,int sum) {
	if(loc == n + 1 || fg) {
		if(sum == k) {//当前总和等于k
			fg = true;
			puts("YES");
			for(int i = 1;i <= n; ++i)
				if(vis[i])
					printf("%d ",a[i]);
			putchar('\n');
		}
		return;
	}
	vis[loc] = 1;
	dfs(loc+1,sum+a[loc]);//选取第loc个位置的数
	vis[loc] = 0;
	dfs(loc+1,sum);//不选去第loc个位置的数
}

int main()
{
	while(~scanf("%d%d",&n,&k)) {
		fg = false;
		for(int i = 1;i <= n; ++i)
			scanf("%d",&a[i]);
		dfs(1,0);
		if(!fg)
			puts("NO");
		memset(vis,0,sizeof vis);
	}
	return 0;
}

 
二进制枚举Code://64MS

#include<bits/stdc++.h>
using namespace std;
int n,k,a[25],b[25];
int main()
{
	while(~scanf("%d%d",&n,&k)) {
		for(int i = 0;i < n; ++i)	scanf("%d",&a[i]);
		bool fg = true;
		for(int i = 0,len = 1 << n;i < len; ++i) {//我们把选取的状态拆分为2^N,然后找到我们需要的就break就行
			long long temp = 0;
			for(int j = 0; j < n; ++j)
				if(i >> j & 1)//选取第j个元素
					b[j] = 1,temp += a[j];
				else//不选去
					b[j] = 0;
			if(temp == k) {
				fg = false;
				puts("YES");
				for(int i = 0;i < n; ++i) {
					if(b[i])
						printf("%d ",a[i]);
				}
				putchar('\n');
				break;
			}
		}
		if(fg)
			puts("NO");
	}
	return 0;
}