洛谷 P1036 【递归】选数

题目描述

已知 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=2

#include<iostream>
#include<math.h>
using namespace std;

int x[20],n,k;//依照题目所设

bool isprime(int n){//判断是否质数
    int s=sqrt(double(n));
    for(int i=2;i<=s;i++){
        if(n%i==0) return false;
    }
    return true;
}
int rule(int left,int sum,int start,int end){
//left为剩余的k，sum为前面累加的和，start和end为全组合剩下数字的选取范围；
//调用递归生成全组合，在过程中逐渐把K个数相加，当选取的数个数为0时，直接返回前面的累加和是否为质数即可
    if(left==0)return isprime(sum);
    int tsum=0;
    for(int i=start;i<=end;i++){
        tsum+=rule(left-1,sum+x[i],i+1,end);
    }
    return tsum;
}
int main(){
    cin>>n>>k;
    for(int i =0;i<n;i++)cin>>x[i];
    cout<<rule(k,0,0,n-1);//调用递归解决问题
}