題目描述
已知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;
}