洛谷 P1036 选数

点击打开链接

题目描述：

已知 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）。

题解：

这题的主要知识点是递归。从n个数中选取k个数，再判断选取的数的和是否为质数。

代码如下：

#include<iostream>
#include<cmath>
using namespace std;

int k,x[25],n;

int is_prime(int n)     //判断质数
{
    if(n<2) return 0;
    if(n==2) return 1;
    for(int i=2;i<=sqrt(n);i++)
        if(n%i==0) return 0;
    return 1;
}

int dfs(int left_count,int sum,int start)   //递归遍历，left_count为剩下的未选的个数，
//sum为当前已选的和，start为劝阻和剩下数字的选取起点，当剩下选取的个数为0时，判断累加和是否为质数
{
    if(left_count==0) return is_prime(sum);
    int _count=0;
    for(int i=start;i<=n-1;i++)
        _count+=dfs(left_count-1,sum+x[i],i+1);
    return _count;
}

int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++) cin>>x[i];
    cout<<dfs(k,0,0);
    return 0;
}