这道题的核心是对给的n个整数中取k个随机组合,并得到和。
主要是用dfs实现的,之前一直不太会用dfs,正好靠这道题练了练dfs。
代码如下:
#include
#include
#include
using namespace std;
int n,k;
int x[20];
bool is_su(int x){
if(x0||x1){
return false;
}
else{
for(int i=2;i<=sqrt(x);i++){
if(x%i0) return false;
}
return true;
}
}
int count=0;
//核心是dfs
void dfs(int num,int sum,int start){
//当已经拿了k个数,判断和是否为素数
if(numk){
if(is_su(sum)){
count++;
}
return ;
}
//从小到大遍历
for(int i=start;i<n;i++){
//这里必须传i+1,不能传start+1!因为每次要取的那个数都是要大于前一个数
dfs(num+1,sum+x[i],i+1);
}
return ;
}
int main(){
cin>>n>>k;
for(int i=0;i<n;i++){
cin>>x[i];
}
dfs(0,0,0);
cout<<count;
}