题目描述
已知 n 个整数b1,b2,…,bn
以及一个整数 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)输出
一个整数(满足条件的方案数)。
样例输入
4 33 7 12 19
样例输出
1 终于刷到dfs了 还是菜啊
/**********************author: yomidate: 18.8.5ps:**********************/#include#include #include using namespace std;int n, k, cnt = 0, sum = 0;int p[50], a[50], b[50];bool vis[50];bool isPrime(int n){ if(n == 1) return false; if(n == 2){ return true; } for(int i=2; i<=sqrt(n); i++){ if(n%i == 0) return false; } return true;}void dfs(int index){ if(index == k+1){ if(isPrime(sum)) cnt++; return; } for(int i=1; i<=n; i++){ if(!vis[i] && i>p[index-1]){ sum+=a[i]; p[index] = i; vis[i] = true; dfs(index+1); sum-=a[i]; vis[i] = false; } } return;}int main(){ memset(vis, false, sizeof(vis)); cin >> n >> k; for(int i=1; i<=n; i++){ cin >> a[i]; p[i] = i; } dfs(1); cout << cnt; return 0;}/**4 33 7 12 19**/