组合+判断素数

题目描述

已知 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);
	}
}