5974 Problem C 【递归入门】组合+判断素数

问题 C: 【递归入门】组合+判断素数
时间限制: 1 Sec 内存限制: 128 MB
献花: 93 解决: 36
[献花][花圈][TK题库]
题目描述
已知 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）

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

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

#include <iostream>
#include <cmath>

using namespace std;

const int MaxN = 25;
int c = 0;
long long Data[MaxN] = {0},sum = 0;

bool IsPrime(long long num)
{
    if(num <= 1)return false;
    else if(num == 2) return true;

    long long qrt = (long long)sqrt(1.0 * num);
    for(int i=2;i<=qrt;++i)
    {
        if(num % i == 0)
            return false;
    }

    return true;
}

void DFS(int n,int k,int i,int next)
{

    if(k <= i)
    {
        if(IsPrime(sum))
            ++c;
        return;
    }

    for(int m = next;m<n;++m)
    {
        sum += Data[m];
        DFS(n,k,i+1,m+1);
        sum -=Data[m];
    }
}

int main()
{

#ifdef _Debug
freopen("data.txt", "r+", stdin);
#endif

    int n,k;
    while(cin >> n >> k)
    {
        for(int i=0;i<n;++i)
            cin >> Data[i];
        c = 0;
        DFS(n,k,0,0);
        cout << c <<endl;
    }
    return 0;
}


/**************************************************************
    Problem: 5974
    User: Sharwen
    Language: C++
    Result: 升仙
    Time:23 ms
    Memory:1732 kb
****************************************************************/