CODEVS-1008 选数

题目描述  Description

已知 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）。

输入描述  Input Description

　键盘输入，格式为：
　　n , k （1<=n<=20，k＜n）
　　x1,x2，…,xn （1<=xi<=5000000）

输出描述  Output Description

屏幕输出，格式为：
　　一个整数（满足条件的种数）。

样例输入  Sample Input

4 3
3 7 12 19

样例输出  Sample Output

1

数据范围及提示  Data Size & Hint

（1<=n<=20，k＜n）
（1<=xi<=5000000）

解题思路：使用深度优先搜索，此题递归的规律很好找，直接贴代码

#include <stdio.h>
#define M 21
 
int n=0,k=0;  //n个整数中选择k个 
int count=0;

int judge(int num);
void arrange(int *num,int current,int address,int sum);

int main(void)
{
	int i=0;
	int num[M]={0};  //num数组接收n个整数
	scanf("%d %d",&n,&k);
	for(i=0;i<n;i++)
	{
		scanf("%d",num+i);
	}
	arrange(num,0,0,0);
	printf("%d",count); 
	return 0;
}
//对num数组进行搜索，赋值加入sum，current为当前确定sum已经加入多少个数
//sum每次加入的值只从num数组中address开始取 
void arrange(int *num,int current,int address,int sum)
{
	int i=0;
	if(current>=k) 
	{
		if(judge(sum)) count++;
		return;
	}
	for(i=address;i<n;i++)
	{
		sum+=num[i];
		arrange(num,current+1,i+1,sum);
		sum-=num[i];
	}
}
//判断num是否为素数 
int judge(int num)
{
	int i=0;
	for(i=2;i*i<=num;i++)
	{
		if(num%i==0) return 0;
	}
	return 1;
}