P1036_选数

最新推荐文章于 2024-10-02 17:08:35 发布

wang_hua_yi

最新推荐文章于 2024-10-02 17:08:35 发布

阅读量253

点赞数

分类专栏：算法洛谷题目代码文章标签：蓝桥杯深度优先算法

本文链接：https://blog.csdn.net/wang_hua_yi/article/details/122725316

版权

洛谷题目同时被 3 个专栏收录

5 篇文章 0 订阅

订阅专栏

算法

2 篇文章 0 订阅

订阅专栏

代码

2 篇文章 0 订阅

订阅专栏

#include <bits/stdc++.h>
using namespace std;
const int N=5000005;
int A[N];//质数数组 
int T[N];//打表 
int X[N];//输入的数字 
int cn=0;
int n,k,sum=0,cnt=0;
void init()
{
	A[cn++]=2;
	int a=2;
	while(1){
		for(int x=a;x<N;x+=a)T[x]=1;
		for(int x=a+1;;x++){
			if(x>=N){
				return;
			}
			if(!T[x]){
				A[cn++]=x;
				a=x;
				break;
			}
		}
	}
}

void dfs(int d,int in){
	if(d==k){
		if(sum<N){
			//cout<<"sum: "<<sum<<endl;
			if(T[sum]==0){//0的位置是质数 
				cnt++;
			}
		}
		return;
	}
	for(int i=in+1;i<n;i++){
		sum+=X[i];
		dfs(d+1,i);
		sum-=X[i];
	}
}


int main() {
	init();//初始化质数表
	//for(int i=0;i<100;i++)cout<<A[i]<<" ";
	//cout<<endl;
	T[0]=1;T[1]=1; 
	for(int i=0;i<cn;i++)T[A[i]]=0;
	//for(int i=0;i<100;i++)cout<<i<<":"<<T[i]<<" ";
	cin>>n>>k;
	for(int i=0;i<n;i++)cin>>X[i];
	dfs(0,-1);
	cout<<cnt<<endl;
	return 0;
}