NOIP 2002 选数【DFS】

NOIP 2002 选数
暴力枚举所有可以发生的情况即可。
试图素数筛，结果发现自己是个弱智= =，并不能优雅起来。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<map>
#include<queue>
#include<vector>
using namespace std;
#define ll long long int
#define INF 0x3f3f3f3f
const int maxn = 25;
int a[maxn];
int p[maxn];
int n, k;
bool judge_prime(int x) {//判断素数的函数
	for (int i = 2; i <= (int)sqrt(1.0*x); i++) {
		if (x%i == 0) return 0;
	}
	return 1;
}
int dfs(int id, int cnt, int sum) {
	//当前枚举下标，当前统计的数字个数，当前统计的数字和
	int res = 0;
	if(cnt == k) return(judge_prime(sum));
	//如果搜索到底层且恰好为素数，返回1；
	if(cnt < k)
		for (int i = id + 1; i <= n; i++) {
			res += dfs(i, cnt + 1, sum + a[i]);
			//加上递归子树的结果。
		}
	return res;
}

int main()
{
	scanf("%d %d", &n, &k);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &a[i]);
	}
	int ans = dfs(0, 0, 0);
	printf("%d\n", ans);
	return 0;
}
//_CRT_SECURE_NO_WARNINGS