【DFS深度优先搜索】选素数

链接：https://ac.nowcoder.com/acm/contest/74168/T
来源：牛客网

题目描述

已知 n 个整数 x1,x2,…,xn，以及一个整数 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

链接

1. 深度优先搜索(DFS)
   总结(算法+剪枝+优化总结)
2. DFS (深度优先搜索) 算法详解 + 模板 +
   例题，这一篇就够了
3. P1036 [NOIP2002 普及组]
   选数

代码

#include <bits/stdc++.h>
using namespace std;

int n, k, cnt = 0;
int a[21];

//判断素数
int prime(int x)
{
	for (int i = 2; i <= sqrt(x); i++)
	{
		if (x % i == 0) return 0;
	}
	return 1;
}

//深度优先搜索
void dfs(int x, int sum, int num)
{
	if (num == k + 1)
	{
		if (prime(sum)) cnt++;
		return;
	}
	for (int i = x; i <= n - k + num; i++)			
	{										
		dfs(i + 1, sum + a[i], num + 1);//整个进程反复进行直到所有节点都被访问为止
	}
}
int main()
{
	scanf("%d %d", &n, &k);
	for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
	dfs(1, 0, 1);
	printf("%d", cnt);
	return 0;
}