P1036 [NOIP2002 普及组] 选数

[NOIP2002 普及组] 选数

题目描述

已知 $n$ 个整数 $x_1,x_2,\cdots ,x_n$，以及 $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\le n\le 20$，$k<n$）。

第二行 $n$ 个整数，分别为 $x_1,x_2,\cdots ,x_n$（$1\le x_i\le 5\times 10^6$）。

输出格式

输出一个整数，表示种类数。

样例 #1

样例输入 #1

4 3
3 7 12 19

样例输出 #1

1

提示

【题目来源】

NOIP 2002 普及组第二题

---

import java.util.Scanner;

public class Main {
	
	static int N = 30;
	static int x[] = new int[N];
	static int n, k, cnt, sum;
	static int PN = 5000010, pcnt;
	static int primes[] = new int[PN];
	static boolean st[] = new boolean[PN];
	
	// 筛法求素数O(n)，只筛一次，查询时O(1)
	public static void get_primes(int pn) {
		for(int i = 2; i <= pn; i++) {
			if(!st[i]) {
				primes[pcnt++] = i;
			}
			for(int j = 0; primes[j] * i <= pn; j++) {
				st[primes[j] * i] = true;
				if(i % primes[j] == 0) break;
			}
		}
	}
	
	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++) {
			x[i] = sc.nextInt();
		}
		get_primes(PN-1);
		
		dfs(1, 1);
		System.out.println(cnt);
		
		sc.close();
	}
	
	// 深搜实现组合型枚举，这里优化了一下，x下标按字典序从小到大，去重
	public static void dfs(int u, int start) {
		if(u > k) {
			if(!st[sum]) cnt++;
			return;
		}
		for(int i = start; i <= n; i++) {
			sum += x[i];
			dfs(u + 1, i + 1);
			sum -= x[i];
		}
	}
}