35 lines
1.0 KiB
C++
35 lines
1.0 KiB
C++
#include <bits/stdc++.h>
|
||
|
||
using namespace std;
|
||
const int N = 30;
|
||
int a[N];
|
||
|
||
//是不是质数
|
||
bool isPrime(int n) {
|
||
for (int i = 2; i <= n / i; i++)
|
||
if (n % i == 0) return false;
|
||
return true;
|
||
}
|
||
|
||
int main() {
|
||
int n, k, ans = 0;
|
||
cin >> n >> k;
|
||
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
|
||
|
||
int U = 1 << n; //U-1即为全集 ，比如 1<<5 就是 2的5次方，就是32，U=32。而U-1=31，就是表示 1 1 1 1 1
|
||
for (int S = 0; S < U; S++) { //枚举所有子集[0,U)
|
||
//找到k元子集
|
||
if (__builtin_popcount(S) == k) {
|
||
int sum = 0;
|
||
//是哪些数存在于子集中呢？
|
||
for (int i = 0; i < n; i++) {
|
||
int bit = (S >> i) & 1;
|
||
if (bit) sum += a[i]; //遍历数字S的每一位，如果不是0，表示这一位上的数字是存在的，需要加进来
|
||
}
|
||
//判断子集元素和是不是素数
|
||
if (isPrime(sum)) ans++;
|
||
}
|
||
}
|
||
cout << ans << endl;
|
||
return 0;
|
||
} |