经典深度搜索算法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;
}