P1036 选数

题目要求是通过输入程序，在输入的n个数里面选择出m个数求和，统计和为素数的情况。

最开始，我设计的是如何考虑他是素数的情况，假如他三个数都为素数，那么得出的结果也坑定一个素数，结果也不出我所料，这是一个坑。

 

emmm，我怀疑了半天还是没有找到新的办法，所以只能走通解，用递归遍历整个数组，然后再一次次判断，但是假如一次次判断的话开辟内存的时间也很麻烦，所以采用check函数来快速判断是否为素数； 

bool isnotPrime[5000001];
int n,m,cot=0;
void checkPrime()
{
    int i,j;
    for(int i=2;i<=5000001;i++)
    {
        for(int j=2;j<=i;j++)
        {
            if(i*j>=5000000)break;
            isnotPrime[i*j]=true;
        }
    }
}

 最后是ac代码：

#include"iostream"
#include"vector"
#include"cstdlib"
#include"algorithm"
using namespace std;
/*int n,m,cot=0;
bool isnotPrime[1000];*/
bool isnotPrime[5000001];
int n,m,cot=0;
void checkPrime()
{
    int i,j;
    for(int i=2;i<=5000001;i++)
    {
        for(int j=2;j<=i;j++)
        {
            if(i*j>=5000000)break;
            isnotPrime[i*j]=true;
        }
    }
}
void dfs(vector<int>&v,int index,int sum,int mark)
{
    if(mark==m)
    {
        if(isnotPrime[sum]==false)cot++;//假如为false，说明是质数，则计数
        return;
    }
    for(int i=index;i<n;i++)
    {
        dfs(v,i+1,sum+v[i],mark+1);
    }
}
int main()
{
    vector<int> v;
    int k;
    cin>>n>>m;
    for(int i=0;i<n;i++)
    {
        cin>>k;
        v.push_back(k);
    }
    checkPrime();
    dfs(v,0,0,0);
    cout<<cot<<endl;
    system("pause") ;
    return 0;
}