DFS深度搜索算法应用

经典深度搜索算法DFS,助力理解和应用DFS

问题出处：
https://www.luogu.com.cn/problem/P1036

题目描述：
已知 n 个整数 x1,x2,…,xn，以及1个整数k(k<n)。从n个整数中任选k个整数相加，可分别得到一系列的和。例如当n=4,k=3,4个整数分别为3,7,12,19时，可得全部的组合与它们的和为：
3+7+12=22
3+7+19=29
7+12+19=38
3+12+19=34
现在，要求你计算出和为素数共有多少种。
例如上例，只有一种的和为素数：3+7+19=29。

输入格式：
键盘输入，格式为：
n,k(1≤n≤20,k<n)
x1,x2,…,xn(1≤xi≤5000000)
输出格式：
屏幕输出，格式为： 1个整数（满足条件的种数）。

样例输入：
4 3
3 7 12 19
样例输出：
1

代码：

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int n,k;
int sum=0;
int nums[25];//存储可选择的全部数
int isPrime(int num)//判断是否是素数
{
	if(num==2)
		return 1;
	int temp=sqrt(num);
	for(int i=2;i<=temp;i++)
	{
		if(num%i==0)
		return 0;
	}
	return 1;
}
void dfs(int index,int count,int num)//index为数组的下标位置，count是已选择的数量，num记录已经选择的所有数的和
{
	if(count==k)
	{
		if(isPrime(num))
		sum++;
		return;
	}
	for(int i=index;i<n;i++)//匹配到所有的组合可能性
	{
		dfs(i+1,count+1,num+nums[i]);
	}
}
int main()
{
	cin>>n>>k;
	for(int i=0;i<n;i++)
		cin>>nums[i];
	dfs(0,0,0);
	cout<<sum<<endl;
	return 0;
}