思路
- 先用线性筛得到所有可能用到的素质,再用DFS判断不同组合之和是否为素数。
代码实现
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int s = 5 * (int)1e6;
vector<int> p;
bool np[s];
int n, k;
bool isVisited[20];
int num[20];
int ans;
int sum;
void pre()
{
for (int i = 2; i <= s; i++)
{
if (!np[i])
{
p.push_back(i);
}
for (auto it : p)
{
if (i * it > s)
{
break;
}
np[i * it] = true;
if (i % it == 0)
{
break;
}
}
}
}
void dfs(int t, int s)
{
if (t == k)
{
if (find(p.begin(), p.end(), sum) != p.end())
{
ans++;
}
return;
}
for (int i = s; i < n; i++)
{
if (!isVisited[i])
{
sum += num[i];
isVisited[i] = true;
dfs(t + 1, i + 1);
sum -= num[i];
isVisited[i] = false;
}
}
}
int main()
{
ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
pre();
cin >> n >> k;
for (int i = 0; i < n; i++)
{
cin >> num[i];
}
dfs(0,0);
cout << ans;
return 0;
}