【洛谷】P1036选数（DFS）

最新推荐文章于 2025-03-22 14:46:23 发布

m0_74758366

最新推荐文章于 2025-03-22 14:46:23 发布

阅读量1.1k

点赞数 30

分类专栏：题解文章标签：深度优先算法图论

本文链接：https://blog.youkuaiyun.com/m0_74758366/article/details/136351282

版权

题解专栏收录该内容

32 篇文章

订阅专栏

这篇文章介绍了如何使用C++编程语言，结合深度优先搜索（DFS）算法和素数判断函数，解决从给定n个整数中选择k个素数，使得它们的和最小的问题。

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

#include<iostream>
#include<math.h>
using namespace std;

const int N = 5000010;

int n, k, x[25], ans = 0, count = 0;
bool v[N] = {0};

bool isPrime(int x)
{
	if(x <= 3) return true;
	
	if(x % 2 == 0) return false;
	
	int q = sqrt(x) + 1;
	for(int i = 3;i <= q; i += 2)
	{
		if(x % i == 0) return false;
	}
	
	return true;
}

// i表示当前在考虑第i个数
// sum表示当前选择的所有数的和
// count表示目前选择了几个数 
void dfs(int i, int sum, int count)
{
	if(count == k && isPrime(sum))
	{
		++ans;
		return;
	}
	
	if(i > n) 
		return; // 选的数量不合适则return 
		
	dfs(i+1, sum+x[i], 1+count); // 选当前的数 
	dfs(i+1, sum, count); // 跳过当前的数 
}

int main()
{
	cin>>n>>k; 
	for(int i=1;i<=n;++i) cin>>x[i];
	
	dfs(1, 0, 0);
	
	cout<<ans;
	
	return 0; 
}