53 lines
1.0 KiB
C++
53 lines
1.0 KiB
C++
#include <bits/stdc++.h>
|
||
|
||
using namespace std;
|
||
const int N = 30;
|
||
int a[N];
|
||
int n, K;
|
||
int cnt;
|
||
|
||
int sum;
|
||
int m;
|
||
|
||
//判断一个数是不是质数
|
||
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 走在第几个箱子面前
|
||
* @return 获取可行方法数
|
||
*/
|
||
void dfs(int step) {
|
||
//脚落地
|
||
if (m == k) {
|
||
if (isPrime(sum)) cnt++;
|
||
return;//已经完成选择k个数，判断和是不是质数.如果是质数，说明找到了一种合法解
|
||
}
|
||
|
||
//这句需在放在后面
|
||
if (step == n + 1) return;
|
||
|
||
//选择当前数字
|
||
sum += a[step];//回溯法
|
||
m++;
|
||
dfs(step + 1);
|
||
m--;
|
||
sum -= a[step];//加多少返多少
|
||
|
||
//放弃当前数字
|
||
dfs(step + 1);
|
||
}
|
||
|
||
int main() {
|
||
cin >> n >> k;
|
||
for (int i = 1; i <= n; i++) cin >> a[i];
|
||
dfs(1);
|
||
cout << cnt << endl;
|
||
return 0;
|
||
}
|