P1036 选数

最新推荐文章于 2023-06-07 22:19:26 发布

原创最新推荐文章于 2023-06-07 22:19:26 发布 · 115 阅读

0 ·

CC 4.0 BY-SA版权

C++/PAT/洛谷OJ 专栏收录该内容

257 篇文章

订阅专栏

本文介绍了一个使用C++实现的程序，该程序通过深度优先搜索（DFS）算法来找出所有可能的组合，这些组合由指定数量的元素构成，并且其总和为素数。程序读取输入文件中的整数集合，并计算出符合要求的组合总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<bits/stdc++.h>
using namespace std;
bool isprime(int a){
	if(a==1) return false;
	for(int i=2;i<=sqrt(a);i++){
		if(a%i==0) return false;
	}
	return true;
}
int n,k;
int cnt=0;
vector<int> ppp,temp;
void dfs(int index,int num,int sum){
	if(num==k){
		if(isprime(sum)==true) cnt++;
		return;
	}
	if(num>k||index>=n) return;
	temp.push_back(ppp[index]);
	dfs(index+1,num+1,sum+ppp[index]);
	temp.pop_back();
	dfs(index+1,num,sum);
}
int main()
{
	freopen("in.txt","r",stdin);
	cin>>n>>k;
	ppp.resize(n);
	for(int i=0;i<n;i++){
		cin>>ppp[i];
	}
	dfs(0,0,0);
	cout<<cnt;
	return 0;
}