【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;
}