洛谷P1036 [NOIP2002 普及组] 选数

最新推荐文章于 2024-03-31 12:50:54 发布

一只小绵羊吖

最新推荐文章于 2024-03-31 12:50:54 发布

阅读量242

点赞数 1

文章标签：深度优先算法图论

本文链接：https://blog.csdn.net/Xgggcalled/article/details/128791515

版权

P1036 [NOIP2002 普及组] 选数

运用DFS

#include<bits/stdc++.h>
using namespace std;

const int maxn=22;
int A[maxn];
int prime[10000];
int cnt=0;
//int ans;
int n,k;
bool isPrime(int n){
	if(n<=1){
		return false;
	}
	int sqr=(int)sqrt(1.0*n);
	for(int i=2;i<=sqr;i++){
		if(n%i==0){
			return false;
		}
	}
	return true;
}
void DFS(int index,int nowK,int sum){
	if(nowK==k){
		if(isPrime(sum)){
			prime[cnt++]=sum;
		}
		return;
	}
	if(index==n){
		return;
	}
	DFS(index+1,nowK,sum);
	DFS(index+1,nowK+1,sum+A[index]);
}

int main(){
	scanf("%d%d",&n,&k);
	
	for(int i=0;i<n;i++){
		scanf("%d",&A[i]);
	}
	
	DFS(0,0,0);
	
//	set<int> st;
//	for(int i=0;i<cnt;i++){
//		st.insert(prime[i]);
//	}
	printf("%d",cnt);
}