洛谷 P1036 选数（C++）

年少挽剑世无双

于 2019-04-21 19:28:44 发布

阅读量3.2k

点赞数 4

分类专栏：洛谷文章标签： P1036 选数洛谷

本文链接：https://blog.youkuaiyun.com/qq_21256635/article/details/89437660

版权

洛谷专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索(DFS)算法来找出所有可能的整数组合，这些组合的元素个数固定，且组合之和为素数。通过递归调用实现DFS，对输入的整数数组进行遍历，检查所有可能的k个数的组合，如果组合的和为素数，则计数器加一。最后返回计数器的值。

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

在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;
int n,k;
int a[25];
bool isprime(int n)			//判断素数
{
	for(int i = 2; i*i < n; i++){
		if(n % i == 0)
			return false;
	}
	return true;
}
int cnt = 0;
int dfs(int start,int num,int sum)
{
	if(num == k){
		if(isprime(sum)){
			cnt++;			//如果选数总个数等于k 如果总和是素数则个数+1
		}
	}
	for(int i = start; i < n; i++){				//从剩下的范围中选数
		dfs(i+1,num+1,sum+a[i]);	//区间开头+1 所选的个数+1
	}
	return cnt;
}
int main()
{
	cin >> n >> k;
	for(int i = 0; i < n; i++){
		cin >> a[i];
	}
	cout << dfs(0,0,0);
	return 0;
}