题目描述 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)
这个题感觉一看时间限制只有1s钟,立刻考虑是不是用打表然后查表的方式来解决,但是看了看数据大小还是放弃了,空间是不够用的,所以我初步想到的利用欧拉筛法求素数打表然后用dfs判断查表的方法叫我抛弃了,之后就尝试了一下普通的DFS和求素数,结果就过了,以后对时间和空间的把握要更强一些.
/*************************************************************************
> File Name: 选数.c
> Author: zhanghaoran
> Mail: chilumanxi@gmail.com
> Created Time: 2015年06月09日 星期二 10时09分32秒
************************************************************************/
#include <stdio.h>
#include <string.h>
int n, k; //n为整数个数,k为任选几个数
int a[21];
int flag[21];
int res = 0;
int check_prim(int n){
int i;
if(n == 1 || n == 0)
return 0;
for(i = 2; i *i <= n; i ++ ){
if(n % i == 0)
return 0;
}
return 1;
}
void dfs(int step, int pos, int sum){
int i;
memset(flag, 0, sizeof(flag));
if(step == k && check_prim(sum)){
res ++;
return ;
}
else if(step == k)
return ;
int vv[21];
for(i = 0; i < n; i ++){
vv[i] = flag[i];
}
for(i = pos + 1; i < n; i ++){
if(!vv[i])
dfs(step + 1, i, sum + a[i]);
}
for(i = 0; i < n; i ++)
flag[i] = vv[i];
}
int main(void){
int i;
scanf("%d%d", &n, &k);
for(i = 0; i < n; i ++){
scanf("%d", &a[i]);
}
dfs(0, -1, 0);
printf("%d",res);
}