P1036 选数（简单dfs递归）

题目地址
题目描述
已知 nn 个整数 x_1,x_2,…,x_nx
1
​ ,x
2
​ ,…,x
n
​ ，以及11个整数kk(k<nk<n)。从nn个整数中任选kk个整数相加，可分别得到一系列的和。例如当n=4,k=3n=4,k=3,44个整数分别为3,7,12,193,7,12,19时，可得全部的组合与它们的和为：

3+7+12=223+7+12=22

3+7+19=293+7+19=29

7+12+19=387+12+19=38

3+12+19=343+12+19=34。

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=293+7+19=29。

输入格式
键盘输入，格式为：

n,kn,k(1 \le n \le 20,k<n1≤n≤20,k<n)

x_1,x_2,…,x_n (1 \le x_i \le 5000000)x
1
​ ,x
2
​ ,…,x
n
​ (1≤x
i
​ ≤5000000)

输出格式
屏幕输出，格式为： 11个整数（满足条件的种数）。

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

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<iostream>
#include<algorithm>
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<string>
#define dd double
#define ll long long
dd PI = acos(-1);
using namespace std;
const ll MAXN = 5e6 + 5;
ll INF = 0xfffffff;
ll n, m;
ll ans[25];
ll re[25];
ll total = 0;
map<ll, ll> ma;

ll prime(ll num1) {
	ll flag = 1;
	if (num1 == 1 || num1 == 0) {
		return 2;
	}
	for (ll i = 2; i <= sqrt(num1); i++) {
		if (num1 % i == 0) {
			flag = 2;
			break;
		}
	}
	return flag;
}

void dfs(ll start, ll sum, ll count) {
	if (count == m) {
		ll z = prime(sum);
		//cout << sum << endl;
		if (z == 1) {
			total++;
		}
		return;
	}
	for (ll i = start; i < n; i++) {
		if (re[i] == 0) {
			re[i] = 1;
			dfs(i + 1, sum + ans[i], count + 1);
			re[i] = 0;
		}
	}
	return;
}

int main() {
	for (ll i = 0; i < 24; i++) {
		re[i] = 0;
	}
	cin >> n >> m;
	for (ll i = 0; i < n; i++) {
		cin >> ans[i];
	}
	dfs(0, 0, 0);
	cout << total << endl;
}