48 lines
1.0 KiB
C++
48 lines
1.0 KiB
C++
#include <bits/stdc++.h>
|
||
using namespace std;
|
||
const int N = 25;
|
||
int a[N]; // n个数字都是什么数字
|
||
int n, k; // 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;
|
||
}
|
||
/*
|
||
测试用例
|
||
4 3
|
||
3 7 12 19
|
||
|
||
答案：
|
||
1
|
||
*/
|
||
void dfs(int u, int c, int sum) {
|
||
if (u == n + 1) { // 递归出口
|
||
if (c == k) { // 只有最终是3个数字的组合才有意义
|
||
if (isPrime(sum))
|
||
cnt++; // 多发现了一个质数
|
||
}
|
||
return;
|
||
}
|
||
if (c < k) { // 数量不足，才可以继续选择数字，否则就没有机会再选择数字了
|
||
// 不要这个数字
|
||
dfs(u + 1, c, sum);
|
||
// 要这个数字
|
||
dfs(u + 1, c + 1, sum + a[u]);
|
||
}
|
||
}
|
||
int main() {
|
||
cin >> n >> k;
|
||
for (int i = 1; i <= n; i++)
|
||
cin >> a[i];
|
||
|
||
dfs(1, 0, 0);
|
||
|
||
cout << cnt << endl;
|
||
return 0;
|
||
} |