【暴力枚举 深搜dfs】[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 普及组第二题

这是第一次接触深度优先搜索dfs

判断素数/质数的函数（直接用）：

简单暴力

int prime(int x)//判断质数的函数
{
    if (x == 2 || x == 3 || x == 5)
        return 1;
    if (x < 2 || x % 2 == 0)
        return 0;
    for (int i = 3; i < sqrt(x); i += 2)//每次加2提高效率
        if (x % i == 0)
            return 0;
    return 1;
}

#include<stdio.h>
#include <math.h>
int n,k;//n、k、数组a设为全局变量(刚开始设为局部变量，后面才改的)，以便dfs函数调用
int cnt=0; //表示和为素数的个数
int a[21];
int prime(int x)//判断质数的函数
{
    if (x == 2 || x == 3 || x == 5)
        return 1;
    if (x < 2 || x % 2 == 0)
        return 0;
    for (int i = 3; i < sqrt(x); i += 2)//每次加2提高效率
        if (x % i == 0)
            return 0;
    return 1;
}
           
void dfs(int count, int sum, int start)//dfs搜索k个数 
{
   
    if (count == k) {//count为已选的数的个数，等于k时停止搜索
        if (prime(sum))//sum 为已选数的和，判断是否为素数
            cnt++;
        return;//刚开始老是忘记！
    }
    //for循环进行下一次的判断，下一次的数组下标就是当前数组下标+1，count也是逐个进行计数,sum每次加的都是a[i]；
 //start为已选数的最左下标，从start开始往后找，，防止漏选重选
    for (int i = start; i < n; i++) 
        dfs(count + 1, sum + a[i], i + 1);
    return;//同上
    //如果i从1开始，就会出现一个数被加了很多次，这里是不能被重复加的
}

 
int main() {
    
    scanf("%d%d", &n, &k);
    for (int i = 0; i<n; i++)
        scanf("%d ", &a[i]);           
    dfs(0, 0, 0); //从0开始搜索
    printf("%d\n", cnt);
    return 0;
}