P1036 选数题解

#include <bits/stdc++.h>

using namespace std;
const int N = 30;
int a[N];
int n, k;

//是不是质数
bool isPrime(int n) {
    if (n < 2) return false;
    for (int i = 2; i <= n / i; i++)
        if (n % i == 0) return false;
    return true;
}


/**
 * 功能:获取可行方法数
 * @param step  走在第几个箱子面前
 * @param sum   已经获得的数字和
 * @param m     已经选择了几个数字
 * @return      获取可行方法数
 */
int dfs(int step, int sum, int m) {
    //边界状态
    if (step == n + 1) return 0;//都走冒了还没有停止，说明不可能有结果了
    if (m == k) return isPrime(sum);//已经完成选择k个数，判断和是不是质数.如果是质数，说明找到了一种合法解

    //解析：
    //1、dfs(step + 1, sum + a[step + 1], m + 1)
    //当前的箱子里面的数字要着，然后走到下一个箱子面前，这样，有三个东西发生了变化
    //分别是 面对的箱子是哪个，数字总和，已选择的总数

    //2、dfs(step + 1, sum, m)
    //当前的箱子里面的数字不要，然后走到下一个箱子面前，这样，有一个东西变生了变化
    //面对的箱子是哪个
    return dfs(step + 1, sum + a[step + 1], m + 1) + dfs(step + 1, sum, m);
}

int main() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) cin >> a[i];
    //之所以把step初始值设置为0，可以理解为提前观察，决策后分支.看好了再走路，而不是走了路再看看好不好走。
    cout << dfs(0, 0, 0);
    return 0;
}

P1036 选数题解

原文：https://www.cnblogs.com/littlehb/p/15006108.html