【递归入门】组合+判断素数

问题 C: 【递归入门】组合+判断素数

时间限制: 1 Sec  内存限制: 128 MB

题目描述

已知 n 个整数b1,b2,…,bn

以及一个整数 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） 
第二行n个整数：x1,x2，…,xn （1<=xi<=5000000） 

输出

一个整数（满足条件的方案数）。 

样例输入

4 3

3 7 12 19

样例输出

1

这道题目单个数据的数值太大，不能用筛素数法，但是总的数据量却很小（1<=n<=20）可以单独判断

#include<stdio.h>#include<string.h>#include<math.h>void dfs(int cur,int i);bool check(int a);int an[25],n,k,cnt,max,sum;bool book[25];int main(){cnt=0,max=-1,sum=0;scanf("%d%d",&n,&k);for(int i=0;i<n;i++){scanf("%d",&an[i]);if(an[i]>max)max=an[i];}dfs(0,0);printf("%d",cnt);return 0;}void dfs(int cur,int i){if(cur==k){if(check(sum))cnt++;return ;}for(i;i<n;i++){if(i>n-(k-cur))return ;if(book[i]==false){book[i]=true;sum+=an[i];dfs(cur+1,i+1);sum-=an[i];book[i]=false;}}}bool check(int a){if(a==2||a==3)return true;if(!(a&1))return false;for(int i=3;i<=sqrt(a);i+=2){if(a%i==0)return false;}return true;}