题目链接
考点:DFS+素数判断
思路
通过st数组储存当前数字是否被使用,cnt计数一共有几个数字被使用,cnt==k
时判断sum是否为素数即可
AC代码
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=23;
int n,k,ans,sum,cnt;
int a[N];
bool st[N];//储存第i个数数是否被使用过
bool pd(int x){
if(x==1||x%2==0) return false;
if(x==2||x==3) return true;
int d=sqrt(x);
for(int i=3;i<=d;i+=2){
if(x%i==0) return false;
}
return true;
}
void dfs(int x){
if( x==n ){
cnt=0,sum=0;
for(int i = 0; i < n; i ++){
if(st[i]) cnt++,sum+=a[i];
}
if(cnt!=k) return;
if(pd(sum)) ans++;
return;
}
st[x] = true;//标记
dfs( x+1 );//递推
st[x] = false;//恢复
dfs( x+1 );//递推
}
int main(){
cin>>n>>k;
for(int i=0;i<n;i++) cin>>a[i];
dfs(0);
cout<<ans;
return 0;
}