Description
已知 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)。
Input
键盘输入,格式为:
n , k (1<=n<=20,k<n)
x1,x2,…,xn (1<=xi<=5000000)
Output
屏幕输出,格式为:
一个整数(满足条件的种数)。
Input example
4 3
3 7 12 19
Output example
1
思路:想到的是用深度优先搜索做的,考虑加上或者不加上一个数的问题,将数组遍历,并考虑数组遍历完的限制条件
#include<iostream>
using namespace std;
int n, k;
int *a = new int[n];
int index = 0;
bool isPrimeNum(int x) //从1到sqrt(x)搜索,缩短时间
{
if (x == 0 || x == 1) return false;
int k = sqrt(x);
for (int i = 2; i <= k; i++)
if (x%i == 0) return false;
return true;
}
int dfs(int pos, int i, int sum) {
if (!pos&&isPrimeNum(sum)) //集齐三个数且满足和为素数,则index++
{
index++;
return 0;
}
if (i == n || !pos) return 0; //限制条件:当加满三个数,或者当数组已经遍历完了,则return 0
if (pos&&i < n) { dfs(pos, i + 1, sum); } //不加a[i]
if (pos&&i < n) { dfs(pos - 1, i + 1, sum + a[i]); } //加上a[i]
return 0;
}
int main() {
cin >> n >> k;
for (int i = 0; i < n; i++) {
cin >> a[i];
}
dfs(k, 0, 0);
cout << index;
}