递推与递归——选数
题目描述
已知 n 个整数 x1,x2,…,xn,以及 1 个整数 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)
x1,x2,…,xn(1 ≤ xi ≤ 5000000)
输出格式
屏幕输出,格式为:1 个整数(满足条件的种数)。
输入输出样例
输入样例
4 3
3 7 12 19
输出样例
1
#include<bits/stdc++.h>
using namespace std;
int a[20],n,k,cnt=0,sum=0;
bool isprime(int x) // 判断素数
{
if(x==1){
return false;
}
else{
for(int i=2;i<=sqrt(x);i++){
if(x%i==0){
return false;
}
}
return true;
}
}
void dfs(int x,int y) // 深搜
{
if(x==k){
if(isprime(sum)){
cnt++;
}
}
else{
for(int i=y;i<n;i++){
sum+=a[i];
dfs(x+1,i+1);
sum-=a[i];
}
}
}
int main()
{
cin>>n>>k;
for(int i=0;i<n;i++){
cin>>a[i];
}
dfs(0,0);
cout<<cnt<<endl;
return 0;
}