洛谷P1036 [NOIP2002 普及组] 选数 （搜索）

n个数中选取k个数，判断这k个数的和是否为质数。

在dfs函数中的状态有：选了几个数，选的数的和，上一个选的数的位置；

试除法判断素数即可；

#include<bits/stdc++.h>
using namespace std;
const int N=5e6+5;
int n,k,a[N],ans;

bool isprime(int x){
    int vi=sqrt(x);
    for(int i=2;i<=vi;i++){
        if(x%i==0) return false;
    }
    return true;
}

void dfs(int cnt,int sum,int last){
    if(cnt==k){
        if(isprime(sum)) ans++;
        return ;
    }
    for(int i=last;i<=n;i++)
        dfs(cnt+1,sum+a[i],i+1);
    return ;
}

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