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;
}

猜你喜欢

转载自blog.csdn.net/qq_52172364/article/details/113091782