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

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

时间限制: 1 Sec 内存限制: 128 MB
提交: 194 解决: 71
[提交][状态][讨论版][命题人:外部导入]
题目描述
已知 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 <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=100010;
const int maxp=5000000;
int add[maxn];
int prime[maxp]; 
bool p[maxp]={0};
int pNum=0;
//采用埃式筛法获取一定范围内的素数
void Find_Prime()
{
    int i,j;
    for(i=2;i<maxp;i++)
    {
        if(p[i]==false)
        {
            prime[pNum++]=i;
            //printf("%d ",prime[pNum]);
            for(j=i+i;j<maxp;j+=i)
                p[j]=true;

        }   
    }   
}
//调用库中生成全排列的方法prev_permutation 
int main()
{
    Find_Prime();
    int n,k,a[20],b[20],i,j=1,count,pNumCount=0;
    scanf("%d%d",&n,&k);
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    memset(add,0,sizeof(add));
    for(i=0;i<k;i++)
        a[i]=1;
    for(i=0;i<n;i++)
        scanf("%d",&b[i]);
    for(i=0;i<k;i++)
        add[0]+=b[i];
    while(prev_permutation(a,a+n))
    { 
        for(i=0;i<n;i++)
        {
            if(a[i])
                add[j]+=b[i];   
        }   
        j++;        
    }
    count=j;
    for(i=0;i<count;i++)
    {
        for(j=0;j<pNum&&add[i]>=prime[j];j++)
        {
            if(add[i]==prime[j])
            {
                pNumCount++;
                break;
            }
        }
    }
    printf("%d\n",pNumCount);
    return 0;
}

此代码超时，是因为此代码计算出一定范围内所有的素数值，并且与组合数的和各自循环作比较来判断是否为素数，算法复杂度高，处理数据多，导致算法超时，应该对各组合数的和直接进行判断，遍历小于等于sqrt(此数)的数判断即可，本代码旨在复习算法。