【洛谷每日一练】P1036 选数

P1036 [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

提示

搜索
深度优先搜索,DFS
素数判断,质数,筛法

【题目来源】

洛谷

思路:
boolean类 vis[] 标记已用数, dfs 嵌套实现深搜(之前的文章【Java算法之dfs 与bfs详解】内有解释, 一绝不是想偷懒一,也可以自己 Debug 一下)

AC代码

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner mc = new Scanner(System.in);
        int n = mc.nextInt();    //总个数
        int m = mc.nextInt();    //取几个数相加
        int[] num = new int[n];
        int q = 0;
        boolean[] vis = new boolean[n];
        for (int i = 0; i < n; i++)
            num[i] = mc.nextInt();
        dfs(n, m, num, 1, vis, q);
        System.out.println(count);

    }

    static int count;      //素数个数
    static int sum = 0;
    //全局变量 count,sum 必须放在dfs 外
    //若是只有一行的深度搜索，将循环次数q 放到函数参数内
    static void dfs(int n, int m, int[] num, int x, boolean[] vis, int q) {
        if (x > m) return;

        for (; q < n; q++) {
            if (!vis[q]) {
                sum = sum + num[q];
                vis[q] = true;

                dfs(n, m, num, x + 1, vis, q);   //递归

                vis[q] = false;
                boolean a = false;
                if (x == m) {     //素数判断
                    for (int j = 2; j < sum; j++) {
                        if (sum % j == 0) {
                            a = true;
                            break;
                        }
                    }
                    if (!a) ++count;
                }
                sum = sum - num[q];
            }
        }
    }
}