P1036 [NOIP2002 普及组] 选数题解-CSDN博客

本文链接：https://blog.csdn.net/m0_72674633/article/details/135954816

题目

已知n个整数 $x_{1},x_{2},\cdots ,x_{n}$ 以及1个整数k（k<n）。从n个整数中任选k个整数相加，可分别得到一系列的和。例如当n=4,k=3,4个整数分别为3,7,12,19时，可得全部的组合与它们的和为：

3+7+12=22 3+7+19=29 7+12+19=38 3+12+19=34

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=29。

输入输出格式

输入格式

第一行两个空格隔开的整数n,k（1≤n≤20，k<n）。

第二行n个整数，分别为 $x_{1},x_{2},\cdots ,x_{n}$ （1≤ $x_{i}$ ≤5× $10^{6}$ ）。

输出格式

输出一个整数，表示种类数。

输入输出样例

输入样例

4 3
3 7 12 19

输出样例

解析

这道题目的难点是如何去重，我们采用的是不降原则。

举个例子：
比如说在6里面随便选5个数，那么选法都是什么呢？
瞎枚举？
12345
12346
前两个还不会弄混，然后很可能就乱了，少点数可能不会乱，但是多了就不好整了
比如说在100里随便选50个数。
1 2 3 4 5 6 7 8 9 10 11 12......
所以我们可以运用不降原则：保证枚举的这些数是升序排列，其实真正的不降原则还可以平
比如 1 2 2 3 3 4......但是请注意这道题也不能平，否则就有重复数字了
拿6个里面选3个举例子
1 2 3
1 2 4
1 2 5
1 2 6
第一轮枚举完毕。
第二个数加一
1 3 ？
这个“？”应该是4，因为是升序排列
1 3 4
1 3 5
1 3 6
接着，就是这样
1 4 5
1 4 6
1 5 6
第一位是1枚举完毕
第一位是2呢?
2 3 4
2 3 5
2 3 6
2 4 5
2 4 6
2 5 6
就是这样的，枚举还是蛮清晰的吧
以此类推.....
3 4 5
3 4 6
3 5 6
4 5 6
然后就枚举不了了，结束。
所以说，这样就可以避免判重了。

代码1

#include<iostream>
using namespace std;
int n,k,a[25];
long long ans;
bool isprime(int a){//判断素数 
	for(int i=2;i*i<=a;i++){
		if(a%i==0){
			return false;
		}
	}
	return true;
}
void dfs(int m,int sum,int s){//m表示现在选择了多少个数，sum表示当前的和，s表示升序排列的情况下，目前选择到了那个数在数组中的序号 
	if(m==k){//递归结束条件 
		if(isprime(sum)){
			ans++;
		}
		return;
	}
	for(int i=s;i<n;i++){//从s开始小于n的情况下，依次加1，进行枚举 
		dfs(m+1,sum+a[i],i+1);//递归调用 
	}
	return;
}
int main(){
	cin>>n>>k;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	dfs(0,0,0);
	cout<<ans;
	return 0;
}

代码2

此题目属于子集枚举类型的题目，可以认为是从一个有n个数字的集合中挑选出一些数字（也就是子集），然后判断该子集是否满足某种性质（其和为质数）。

首先考虑如何枚举由n个元素组成的集合中含有k个元素的子集，如果用n位二进制数表示子集，那么目标就是找到所有恰好只有k个1的二进制数。统计二进制中1的个数可以用内建函数__builtin_popcount()，它能直接返回一个数二进制下1的个数；当然也可以逐位分解或者逐位确认。

在找到了一个含有k个元素的子集后，需要把它表示的子集里的所有数提取出来：简单来说就是检查一下每个数是否属于这个集合，利用子集枚举的技巧。剩下的事情就是把数字加起来判断它是否为质数即可。

#include<iostream>
#include<cstdio>
using namespace std;
int a[30];
bool check(int x){//检查质数
	for(int i=2;i*i<=x;i++){
		if(x%i==0){
			return 0;
		}
	}
	return 1;
}
int main(){
	int n,k,ans=0;
	cin>>n>>k;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	int U=1<<n;//U-1即为全集
	for(int S=0;S<U;S++){//枚举所有子集
		if(__builtin_popcount(S)==k){//找到k元子集
			int Sum=0;
			for(int i=0;i<n;i++){
				if(S&(1<<i)){//检查元素是否在集合中
					Sum+=a[i];
				}
			}
			if(check(Sum)){//检查是否为质数
				ans++;
			}
		}
	}
	cout<<ans;
	return 0;
}