cogs 50. [NOIP2002] 选数

50. [NOIP2002] 选数

★   输入文件：choose.in   输出文件：choose.out   简单对比
时间限制：1 s   内存限制：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）

[输出]：
格式为：一个整数（满足条件的种数）。

[输入输出样例]:
输入：choose.in
4 3
3 7 12 19
输出：choose.out
1

思路：深搜。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 100
using namespace std;
int n,k,ans;
int num[MAXN];
bool judge(int now){
    for(int i=2;i<=sqrt(now);i++)
        if(now%i==0)    return false;
    return true;
}
void dfs(int tot,int sum,int step){
    if(tot==k){
        if(judge(sum))    ans++;
        return ;
    }
    for(int i=step;i<=n;i++)
        dfs(tot+1,sum+num[i],i+1);
}
int main(){
    freopen("choose.in","r",stdin);
    freopen("choose.out","w",stdout);
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%d",&num[i]);
    dfs(0,0,1);
    cout<<ans;
}