玄关求调(有注释)
查看原帖
玄关求调(有注释)
1069781
Lzj_qwq楼主2024/8/27 11:45
#include<cstdio>
int n,k,ans,a[5000001];
bool p[10000001]={1,1};
void prime(int n){//埃筛
	for(int i=2;i*i<=n;++i)
		if(p[i]==false)
			for(int j=i*2;j<=n;j+=i)
				p[j]=true;
	return ;
}
void dfs(int flag,int sum){
	if(flag>k){//已累加k个数
		if(p[sum]==false) ++ans;//判定素数
		return ;//回溯
	}
	for(int i=flag;i<=n;++i){
		sum+=a[i];//累加
		dfs(flag+1,sum);//进入下一层
		sum-=a[i];//取消累加
	}
	return ;
}
int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i)
		scanf("%d",&a[i]);
	prime(10000001);
	for(int i=1;i<=n-k+1;++i)//拆分成1~n,2~n...k~n进行处理
		dfs(1,0);
	printf("%d",ans);
	return 0;
}
2024/8/27 11:45
加载中...