C++ 火柴棒问题 (深度优先搜索+剪枝)
有一个由火柴棒作为边组成的N×N 的格子。(N≤5)按照下图,给火柴棒编号。(这是 N=3 的情况,其它情况类似)
https://res.jisuanke.com/img/upload/20180622/4b5a2151fbd2e8151cec08b0920cddc1e3182109.png 这是此图链接
现在将移除某些火柴棒的状态作为初始状态,需要再移除一些火柴棒,以保证图中一个正方形也没有。请求出所有需要移除火柴棒的最少根数。
输入格式
第一行一个整数 T 表示测试数据个数。
对于每个测试数据,第一行整数 NN ,表示正方形的边长。第二行以一个整数 m 开头(m≤2n(n+1)),接下来 m 个不相同的整数,范围在 [1,2n(n+1)] 之间,以空格分开。
输出格式
对每个测试数据,输出一个整数表示最少需要移除的火柴棒个数。
输出时每行末尾的多余空格,不影响答案正确性
要求使用「文件输入输出」的方式解题,输入文件为 matchstick.in,输出文件为 matchstick.out
样例输入
2
2
0
3
3 12 17 23
样例输出
3
3
题解见下,(有注释),此题用深度优先搜索+剪枝,代码比较的复杂,但我保证我的代码时间复杂度很低:
/*
如果移除当前火柴棒后,不能破坏任何正方形,那就没必要移除.cpp
如果把剩下的火柴棒全部移除,也会留下某些正方形,那后面情况就不用考虑了.cpp
*/
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<vector>
#include<utility>
using namespace std;
typedef long long ll;
const int N=90;
vector<int>d[N];
bool belong[N][N]; // belong[i][j] 表示火柴棒 i 是否属于正方形 j.cpp
int mmax[N]; // mmax[i] 表示正方形 i 最大编号的火柴棒.cpp
int m,ct; // m 为总的火柴棒数量,ct 为初始正方形数量.cpp
int ans; // 最终答案.cpp
bool vis[N]; // 火柴棒初始状态 true 表示不存在.cpp
// u为当前火柴棒的编号,sum为当前移除的火柴棒数量,state为当前完整正方形的集合.cpp
int hstar(int u, ll state) {
vector<pair<int, int> > ps;
for (int i = 0; i < ct; i++) {
if (state>>i&1) {
int num = 0; // 火柴棒数量
for (int k=0;k<d[i].size();k++){
int x=d[i][k];
if (x >= u) {
num++;
}
}
ps.push_back(make_pair(num,i));
}
}
sort(ps.begin(), ps.end());
int res =