洛谷-1036 选数

题目描述
已知 n 个整数 x1,x2,…,xn，以及1个整数k(k<n)。从n个整数中任选kkk个整数相加，可分别得到一系列的和。例如当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

解释：写个暴力枚举K个就行了。。在判断是否为素数就OK。

#include <iostream>
#include<cstdio>
#define N 100000001
using namespace std;
int a[22]={0};
long long sum=0;
int num=0;
int n=0,k=0;
int ret=0;
bool ok(long long x){
    if(x==2) return 1;
    if(x%2==0) return 0;
    for(int i=2;i*i<=x;i++){
        if(x%i==0) return 0;
    }
    return 1;
}
void dfs(int pos){
    if(pos==n+1){
        if(ok(sum)) ret++;
        return;
    }
    if(num+n-pos+1>k) dfs(pos+1);
    if(num>=k) return;
    sum+=a[pos];
    num++;
    dfs(pos+1);
    sum-=a[pos];
    num--;
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    dfs(1);
    printf("%d\n",ret);
    return 0;
}