codeup 问题 C: 【递归入门】组合+判断素数
题目描述
已知 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 3
3 7 12 19
样例输出
1
终于刷到dfs了 还是菜啊
/********************** author: yomi date: 18.8.5 ps: **********************/ #include <iostream> #include <cmath> #include <cstring> 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 3 3 7 12 19 **/