1007: 选数（TJOJ）

1007: 选数

时间限制: 1 Sec   内存限制: 128 MB
提交: 4621   解决: 35
[ 提交][ 状态][ 讨论版]

题目描述

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

输出

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

样例输入

4 3
3 7 12 19

样例输出

1

代码实现

#include<stdio.h>
#include<math.h>
int n,k,ans=0,b[20];
  
int is_prime(int x)  //判断是否为素数 
{  
    if(x==0||x==1) 
	return 0;  
    int k=sqrt(x);  
    for(int i=2;i<=k;i++)  
        if(x%i==0) 
            return 0;  
    return 1;  
}  
void dfs(int pos,int i,int sum) 
{  
    if(!pos) 
	{
		if(is_prime(sum)) 
			ans++;
	}  
    else  
        for(i;i<n;i++) 
		dfs(pos-1,i+1,sum+b[i]);  
}  
int main()  
{  
    scanf("%d%d",&n,&k);  
    for(int i=0;i<n;i++)  
        scanf("%d",&b[i]);  
    dfs(k,0,0);  
    printf("%d\n",ans);  
    return 0;  
}