问题 C: 【递归入门】组合+判断素数

题目描述
已知 n 个整数b1,b2,…,bn

以及一个整数 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）
第二行n个整数：x1,x2，…,xn （1<=xi<=5000000）

输出
一个整数（满足条件的方案数）。

样例输入 Copy
4 3
3 7 12 19
样例输出 Copy
1

在做这道题的时候没有用book数组进行标记，因为题目中“第二行n个整数：x1,x2，…,xn（1<=xi<=5000000）” 这些整数中 没有说明不重复，因此可能有相等的两个数，再用book标记这个整数是否被访问就会出错。

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int a[21];
int n,k;
int c[21];
long long int s=0;
int sum=0;
int sushu(int n)
{
    int i;
    if(n==0||n==1)return 0;
    for(i=2; i<=sqrt(n); i++)
    {
        if(n%i==0)return 0;
    }
    return 1;
}
void dfs(int step,int l)
{
    int i;
    if(step==k+1)
    {
        s=0;
        for(i=1; i<=k; i++)
        {
            s=s+c[i];
        }
        if(sushu(s))
        {
            sum++;
        }
        return;
    }
    for(i=l; i<=n; i++)
    {
        c[step]=a[i];
        dfs(step+1,i+1);
    }
}
int main()
{
    int i;
    scanf("%d %d",&n,&k);
    for(i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
    }
    dfs(1,1);
    printf("%d\n",sum);
    return 0;
}