已知 n 个整数 x1,x2,…,x_nx ,以及1个整数k(k<nk<n)。从nn个整数中任选kk个整数相加,可分别得到一系列的和。例如当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
输出格式
屏幕输出,格式为: 1个整数(满足条件的种数)。
输入输出样例
输入
4 3
3 7 12 19
输出
1
#include <bits/stdc++.h>
using namespace std;
int ans=0, n, k, a[20];
bool prime(int x) {
int i;
for (i = 2; i <= floor(sqrt(x)); i++) {
if (x%i == 0) {
return false;
}
}
return true;
}
//上面判素数
void rec(int start, int count, int sum) {
//rec:递归英文recursion缩写
//start:开始选数的地方,是指每组数的第一个数
//count:已经选了几个数
//sum:到目前为止选的数的和
int i;
if (count == k && prime(sum)) {
ans++;
//如果已经选了k个数,判断和是不是素数
//ans是符合条件的和的个数
}
for (i = start; i <= n; i++) {
//从开始选数的地方到n
//每重循环都是一种可能性
//例如:选第二个数,start=2,n=4,有三种可能性,调用自身3次
rec(i + 1, count + 1, sum + a[i]);//不能写递增符号++
//参数1:从a里面下一个数开始选
//参数2:已经选的数的个数+1
//参数3:sum加上这次选的数
//因为直接从下一个数开始选,所以不可能选到之前选过的数
//无需判断当前的数是否被选过
}
}
int main() {
int i;
cin >> n >> k;
for (i = 1; i <= n; i++) {
cin >> a[i];
}
rec(1, 0, 0);
//从第一个数开始找,已经找了0个数,目前的和是0
cout << ans << endl;
return 0;
}