洛谷 P1036 [NOIP2002 普及组] 选数

题目链接

考点：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;
}