You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
python/TangDou/Dfs/DFS4_XuanShua_ByBox_1.cpp

67 lines
1.6 KiB

2 years ago
#include <bits/stdc++.h>
using namespace std;
const int N = 30;
const int INF = 0x3f3f3f3f;
int a[N];
bool st[N];
int n, k;
int cnt;
vector<int> path;
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 start
*
* 3+7+19
* 3+19+7
* 7+3+19
* 7+19+3
* 19+3+7
* 19+7+3 61
*
* @return
*/
void dfs(int step, int sum, int start) {
//当走到第k+1个虚拟箱子面前表示前面k个箱子已经装完
if (step == k + 1) {
if (isPrime(sum)) {
cnt++;
for (int i = 0; i < path.size(); i++)
cout << path[i] << " ";
cout << endl;
}
return;
}
//在没有走完箱子时,考虑当前箱子放哪张卡片
for (int i = start; i <= n; i++) {
if (!st[i]) {
st[i] = true;
path.push_back(i);
dfs(step + 1, sum + a[i], i + 1);
path.pop_back();
st[i] = false;
}
}
}
/*
4 3
3 7 12 19
1
*/
int main() {
cin >> n >> k;
for (int i = 1; i <= n; i++) cin >> a[i];
dfs(1, 0, 1);
cout << cnt << endl;
return 0;
}