题目
已知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个整数,分别为(1≤≤5×)。
输出格式
输出一个整数,表示种类数。
输入输出样例
输入样例
4 3
3 7 12 19
输出样例
1
解析
这道题目的难点是如何去重,我们采用的是不降原则。
举个例子:
比如说在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;
}