P1036 选数题解

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 选数题解

上一篇:计划任务


下一篇:vue实现div高度可拖拽