题目描述
已知 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
话不多说,直接上代码, 代码中都有解释
import java.util.Scanner;
public class zuhesushu2 {
static int sum = 0, n, k,count = 0; //sum代表从某个组合之和,若该和为素数,count++,n个数选k个
static int[] arr = new int[5000001];//arr表示first数组某个全排列的下标
static int[] first = new int[5000001];//first中存放要选的n个数,但其大小不为n,first下表存放在arr中;
public static boolean isPrime(int n) {//判断是否为质数
if(n<2) return false;
if(n == 2) return true;
if(n % 2 == 0) return false;
for(int i = 3;i <= Math.sqrt(n); i += 2)
if(n%i == 0)
return false;
return true;
}
public static void dfs2(int index) {//进行搜索,index代表搜索的深度
if(index == k+1) {
sum = 0;//记得给sum初始化
for(int i = 1;i <= k;i++) {//arr里存放first数组组合后的下标
sum += first[arr[i]];
}
if(isPrime(sum))//判断是否为素数
count++;
return;
}
for(int i = arr[index-1]+1;i <= n;i++) {//i = arr[index-1]+1保证后一个数大于前一个数,即保证字典序
arr[index] = i;//arr数组其实只需要k+1个大小就行
dfs2(index+1);//进行递归
}
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
k = sc.nextInt();
for(int i = 1;i <= n;i++) {//从1开始赋值,为了匹配自然序,如果想从0开始赋值,只需将first[arr[i]]改为first[arr[i]-1],然后在函数中添加一个first数组参数
int temp = sc.nextInt();
first[i] = temp;
}
dfs2(1);//从1开始搜索,因为要在dfs中保证前一个数小于后一个数防止数组越界,同时也匹配自然序
System.out.println(count);
}
}