python/TangDou/LuoGuBook/P1036_dfs.cpp

#include <bits/stdc++.h>

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

//判断一个数是不是质数
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      获取可行方法数
 */

void dfs(int step, int sum, int m) {
    //脚落地
    if (m == k) {
        if (isPrime(sum)) cnt++;
        return;//已经完成选择k个数，判断和是不是质数.如果是质数，说明找到了一种合法解
    }

    //这句需在放在后面
    if (step == n + 1) return;

    //选择当前数字
    dfs(step + 1, sum + a[step], m + 1);

    //放弃当前数字
    dfs(step + 1, sum, m);
}

int main() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) cin >> a[i];
    dfs(1, 0, 0);
    cout << cnt << endl;
    return 0;
}