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

问题 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;
}