【02NOIP普及组】选数

题目描述
已知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)

第二行为n个数

x1 x2 ……xn(1≤xi≤5000000),各数之间用一个空格隔开)

输出格式
一个整数(满足条件的种数)。

输入输出样例
输入 #1 复制
4 3
3 7 12 19
输出 #1 复制
1

dfs解决问题 dfs dfs dfs dfs dfs

#include<bits/stdc++.h>
using namespace std;
#define N 22
#define M 100000001
int a[N], n, k, ans = 0;
bool f[M];
bool is_prime(int x){
	for(int i=2; i<=sqrt(x); i++){
		if(x%i == 0){
			return false;
		}
	}
	return true;
}
void dfs(int s, int c, int p){//s:已选数的和 c:已选c个数 p:已选到了数组p下标的数 
	if(c == k){
		if(is_prime(s)) {
			++ans;
		}
		return ;
	}
	for(int i=p+1; i<=n; i++){
		dfs(s+a[i], c+1, i);
	}
	return ;
}
int main(){
	cin >> n >> k;
	for(int i=1; i<=n; i++)
		cin >> a[i];
	dfs(0,0,0);
	cout << ans;
	return 0;
}