洛谷P1036 Java

题目出处点这里

思路见代码：

package reintroduction_recursion;

import java.util.ArrayList;
import java.util.Scanner;

public class P1036 {
	
	static int n;
	static int k;
	static ArrayList<Integer> list;//存储满足组合的数
	static int res = 0;//结果
	static boolean[] isUsed;//判断相应位置有没有被用过
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		k = sc.nextInt();
		list = new ArrayList<Integer>();
		isUsed = new boolean[n];
		int arr[] = new int[n];
		for (int i = 0; i < n; i++) {
			arr[i] = sc.nextInt();
		}
		dfs(arr,0,0);
		System.out.println(res);
	}
	
	//dfs
	/**
	 * 
	 * @param arr 待取的数组
	 * @param has 目前取了几个数
	 * @param now 现在取的数所在的下标
	 */
	public static void dfs(int[] arr, int has, int now) {
		if (has == k) {
			if (isPrime(SUM())) {
				res++;
			}
			return;
		}
		for (int i = now; i < n; i++) {//如果i=0就会每次从头搜索起变成了排列,i=now每次搜索下标之后的数,前面的不用搜索
			if (!isUsed[i]) {//没有用过
				list.add(arr[i]);//先加进去
				isUsed[i] = true;//再置为用过
				dfs(arr,has+1, i);//开始下一次递归，同时已有数has+1,此时所在数的下标就是now=i
				list.remove(list.size()-1);//去掉最后一个数
				isUsed[i] = false;//并把去掉的数置为没用过
			}
		}
	}
	
	//判断质数
	public static boolean isPrime(int sum) {
		for (int i = 2; i <= Math.sqrt(sum); i++) {
			if (sum % i == 0) {
				return false;
			}
		}
		return true;
	}
	
	//求和
	public static int SUM() {
		int sum = 0;
		for (int i = 0; i < list.size(); i++) {
			sum = sum + list.get(i);
		}
		return sum;
	}
}