题目描述:
已知 nnn 个整数 x1,x2,…,xnx_1,x_2,…,x_nx1,x2,…,xn,以及111个整数kkk(k<nk<nk<n)。从nnn个整数中任选kkk个整数相加,可分别得到一系列的和。例如当n=4,k=3n=4,k=3n=4,k=3,444个整数分别为3,7,12,193,7,12,193,7,12,19时,可得全部的组合与它们的和为:
3+7+12=223+7+12=223+7+12=22
3+7+19=293+7+19=293+7+19=29
7+12+19=387+12+19=387+12+19=38
3+12+19=343+12+19=343+12+19=34。
现在,要求你计算出和为素数共有多少种。
例如上例,只有一种的和为素数:3+7+19=293+7+19=293+7+19=29。
输入格式
键盘输入,格式为:
n,kn,kn,k(1≤n≤20,k<n1 \le n \le 20,k<n1≤n≤20,k<n)
x1,x2,…,xn(1≤xi≤5000000)x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1,x2,…,xn(1≤xi≤5000000)
输出格式
屏幕输出,格式为: 111个整数(满足条件的种数)。
输入输出样例
输入 #1
4 3
3 7 12 19
输出 #1
1
解法:
import java.io.BufferedReader;
import java.io.InputStreamReader;
public class Main {
public static void main(String args[]) throws Exception {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
String line = bf.readLine();
String[] sp = line.split(" ");
int n = Integer.parseInt(sp[0]);
int k = Integer.parseInt(sp[1]);
int[] arr = new int[n];
line = bf.readLine();
sp = line.split(" ");
for (int i = 0; i < arr.length; i++) {
arr[i] = Integer.parseInt(sp[i]);
}
int[] count = {0};
backtrack(0, -1, k, n, 0, count, arr);
System.out.println(count[0]);
}
private static void backtrack(int x, int y, int k, int n, int num, int[] count, int[] arr) {
if (x == k) {// x代表元素的个数 y代表从哪个位置开始添加元素
if (isPrime(num)) {
count[0]++;
}
return;
}
for (int i = y + 1; i < n; i++) {
num += arr[i];
backtrack(x + 1, i, k, n, num, count, arr);
num -= arr[i];
}
}
private static boolean isPrime(int x) {
for (int i = 2; i * i <= x; i++) {
if (x % i == 0) {
return false;
}
}
return true;
}
}