洛谷 1036 选数 搜索剪枝

https://www.luogu.org/problemnew/show/P1036

题目描述

已知 n 个整数 x1,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≤n≤20,k<n)

x1,x2,…,xn(1≤xi≤5000000)

 

输出格式：

 

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

 

输入输出样例

输入样例#1： 复制

4 3
3 7 12 19

输出样例#1： 复制

1

思路：在不会搜索的时候做这道题emmm……现在看来应该很简单了，n个数，每个数只有选或不选两种情况，当选上的数达到k个的时候，我们判断和是否为素数即可。当然不是所有的情况都需要进行判断的，我们假设当前要选第i个数，已经选了cur个数，(总共有n个数 要选k个数)那么当cur>k或者cur+n-i+1<k的时候就不用再继续搜索了，因为这时候已经选了超过k个数或者把剩下的数都选上也达不到k个数，必定不满足题意。

#include<iostream>
#include<cstdio>
#include<stack>
#include<cmath>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f3f
#define EPS 1e-10
using namespace std;

int n,k;
int cnt=0;
int a[24];

int prime(int n)
{
    if(n==1)
        return 0;
    if(n==2||n==3)
        return 1;
    for(int i=2;i*i<=n;i++)
        if(n%i==0)
            return 0;
    return 1;
}

void dfs(int sum,int i,int cur);

int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    dfs(0,1,0);
    printf("%d\n",cnt);
    return 0;
}

void dfs(int sum,int i,int cur)
{
    if(cur==k)
    {
        if(prime(sum))
            ++cnt;
    }
    if(cur>=k||cur+n-i+1<k)
        return ;
    dfs(sum+a[i],i+1,cur+1);
    dfs(sum,i+1,cur);
}