2021ACM俱乐部后备营个人训练赛第12场（问题 B: 选数1 dfs回溯

问题 B: 选数1
时间限制: 1 Sec 内存限制: 128 MB

题目描述
已知 n 个整数 x1,x2,…,xn，以及一个整数 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）
x1,x2，…,xn （1<=xi<=5000000）
输出
屏幕输出，格式为：
一个整数（满足条件的种数）。
样例输入 Copy
4 3
3 7 12 19
样例输出 Copy
1

思路
搜搜搜 搜够k个判断 哎哎哎啊 就这
我超了 真的 没话说

#include<bits/stdc++.h>
int cnt=0;
using namespace std;
typedef long long ll;
const int maxn=1e3+199;
int k,nn;
int sum=0;
int pan(ll n)
{
    for(int i=2;i<=sqrt(n);i++)
    {
        if(n%i==0)
            return 0;
    }
    return 1;
}

void dfs(int n,int m)
{
    if(m==k)
    {
        if(pan(bn))
            sum++;
            return;
    }
    if(n>nn)
        return;
    bn+=a[n];
    dfs(n+1,m+1);
    bn-=a[n];
    dfs(n+1,m);
}
int main(){

  cin>>nn>>k;
   for(int i=1;i<=nn;i++)
    cin>>a[i];
  dfs(1,0);
   cout<<sum;
	return 0;
}