选数#dfs#luogu

# [NOIP2002 普及组] 选数

## 题目描述

已知 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≤n≤20，k<n）。

第二行 n 个整数，分别为x1​,x2​,⋯,xn​（1≤xi​≤5×106）。

## 输出格式

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

## 样例 #1

### 样例输入 #1

```
4 3
3 7 12 19
```

### 样例输出 #1

```
1
```

## 提示

----------------------------------------------------------------------------------------------------

有一段时间没写算法了，哎，生疏了，这道题原本给的算法提示是bfs但是这里我用了dfs,浅浅讲一下dfs和bfs的区别

手画的，担待

#include<stdio.h>
#include<math.h>
//设置全局变量的话后面轻松一点
int arr[25], brr[25];
int n, k;
int num = 0, sum = 0, dam = 0;
int prim(int sum) {
//比较常规的求素数方法，sqrt()是平方根函数，得出结果是sum的平方根
    for (int i = 2; i < sqrt(sum); i++) {
        if (sum % i == 0) {
            return 0;
        }
    }
    return 1;
}
//好了，可以开始深度优先搜索了
void dfs() {
    for (int i = 0; i < n; i++) {
        if (brr[i] == 0) {
            sum = sum + arr[i];
            num++;
            brr[i] = 1;
            if (num < k) {
                dfs();
                //这里不能只写brr[i]=0;这么处理是为了让他每个结果只出现一遍不至于重复
                //如果不这么写，会发现后面，很多重复结果，至于原因，自己查
                for (int j = i+1; j < n; j++) {
                    brr[j] = 0;
                }
                sum = sum - arr[i];
                --num;
                continue;
                //这里必须加continue！！！不能加break,不加的话后面num--,都会乱
            }
            else if (num == k) {
                dam = dam + prim(sum);
            }
            brr[i] = 0;
            sum = sum - arr[i];
            --num;
        }
        
    }
}
int main() {
    scanf("%d %d", &n, &k);
    for (int i = 0; i < n; i++) {
        scanf("%d", &arr[i]);
    }
    dfs();
    printf("%d", dam);
}