洛谷1036 选数（DFS)

已知 $n$ 个整数 x1,x2,…,xnx_1,x_2,…,x_nx1​,x2​,…,xn​ ，以及 $1$ 个整数 $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\le n\le 20,k<n$ )

x1,x2,…,xn(1≤xi≤5000000)x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1​,x2​,…,xn​(1≤xi​≤5000000)

输出格式：

屏幕输出，格式为： $1$ 个整数（满足条件的种数）。

输入输出样例

输入样例#1： 复制

4 3 3 7 12 19

输出样例#1： 复制

1

思路很简单:DFS枚举出各种情况，再加上个判断素数就好了。

下面贴上代码

#include <iostream>
#include <cmath>
#include <algorithm>
#include <iomanip>
#include <cstring>
using namespace std;
typedef long long ll;
ll n,k,A[25],num;
int ch(ll x)
{
    for(int i=2;i<=sqrt(x);i++)
    {
        if(x%i==0)
            return 0;
    }
    return 1;
}
void dfs(ll a,ll b,ll c)
{
     if(b==n+1)    //判断出界
        return ;
     if(c==k)    //判断素数
     {
         if(ch(a)==1)
         {
             num++;
             //cout<<a<<endl;
         }
        return ;
     }
     for(int i=b+1;i<=n;i++) //枚举
        dfs(a+A[i],i,c+1);
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>k;
    for(int i=1;i<=n;i++)
        cin>>A[i];
    sort(A+1,A+n+1);
    dfs(0,0,0);
    cout<<num<<endl;
    return 0;
}