暴力枚举--选数

题目描述

已知 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

这种升序排列可以避免重复，无需去重便可以遍历所有可能