题目描述
已知 n 个整数 x1,x2,⋯,xn,以及 1 个整数 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≤5×10^6)。
输出格式
输出一个整数,表示种类数。
输入输出样例
输入
4 3 3 7 12 19
输出
1
import java.util.Scanner;
public class Main{
static int k,n;
static long count = 0;
static int[] arr ;
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
n = scanner.nextInt();
k = scanner.nextInt();
arr = new int[n];
for(int i = 0;i < n;i++) {
arr[i] = scanner.nextInt();
}
dfs(0,0,0);
System.out.print(count);
//System.out.print(judge(29));
}
public static boolean judge(long i) {//判断是否是素数
boolean flag = true;
if(i == 1) return false;
for(long j = 2;j <= Math.sqrt(i);j++) {
if(i % j == 0) flag = false;
}
return flag;
}
public static void dfs(int num,long sum,int begin) {
//num是当前找的数的个数;sum是这些数之和;begin升序排列的初始值
if(num == k) {//找够k个数就判断是否为素数
if(judge(sum)) count++;
return;
}
for(int i = begin;i < n;i++) {
dfs(num + 1,sum + arr[i],i + 1);
//i+1防止算重
}
return;//代表找完了,深搜结束
}
}
升序排列:假如有4个数1,2,3,4,从中选3个数求和,有以下情况:
1+2+3
1+2+4
1+3+4
2+3+4
这种升序排列可以避免重复,无需去重便可以遍历所有可能