组合+判断素数

题目描述

 

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

 

样例输出 

 

1

代码

#include <bits/stdc++.h>

using namespace std;

int n,m,a[30],b[69],j;

bool used[30];

bool is(int x)

{

    if(x<=1) return false;

    for(int i=2;i<=sqrt(x);i++)

    {

        if(x%i==0) return false;

    }

    return true;

}

void dfs(int k)

{

    if(k>m)  

    {

        int cnt=0;

        for(int i=1;i<=m;i++)    cnt=cnt+b[a[i]];

        if(is(cnt)) j++;

        return;

    }

    for(int i=1;i<=n;i++)

    {

        if(used[i]==false&&i>a[k-1])

        {   a[k]=i; 

            used[i]=true;

            dfs(k+1);

            used[i]=false;

        }       

    }

    }

int main()

{

    cin>>n>>m;

    for(int i=1;i<=n;i++)    

        cin>>b[i];

    dfs(1);

    cout<<j;

    return 0;

}