C++ 火柴棒问题 (深度优先搜索+剪枝)

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 = 0;
    vector<bool> used(m, false);    // 已选集合中是否含有这根火柴棒
    for (int i = 0; i < ps.size(); i++) {
        int id = ps[i].second;      // 正方形编号
        bool ok = true;
        for (int k=0;k<d[id].size();k++) {
            int x=d[id][k];
            if (used[x]) {
                ok = false;
                break;
            }
        }
        if (ok) {
            res++;
            for (int k=0;k<d[id].size();k++) {
                int x=d[id][k];
                used[x] = true;
            }
        }
    }
    return res;
}
void dfs(int u,int sum,ll state){
    if (sum + hstar(u,state)>= ans) {
    	return;
	}
    if (u==m+1){
        if (state==0){
            ans=min(ans,sum);
        }
        return;
    }
    // 初始火柴棒不存在,直接搜下一个.cpp
    if (vis[u]){
        dfs(u+1,sum,state);
        return;
    }
    int flag=0;	// 0 不用移除  1 可以移除  2 必须移除.cpp
    for (ll i=0;i<ct;i++){
        if ((state & (1<<i)) && belong[u][i]){
            if (mmax[i]==u){
                flag=2;
            }else if (flag==0){
                flag=1;
            }
        }
    }if (flag==0){
        dfs(u+1,sum,state);
    }else if (flag==2){
        for (int i=0;i<ct;i++){
            if ((state>>i&1) && belong[u][i]){
                state^=1LL<<i;
                /*^ 是把两个数写成二进制的样子,对每一位而言,相同的话结果就是 				0,不同结果就是 1这里 state ^= 1LL << i; 相当于把 state 的第 i 位,0 变 1 或者 1 变 0,这里它肯定是 1 变 0,>> 是右移,相当于是去掉二进制最后多少位,state >> i & 1,就是看第 i 位是不是 1*/
            }
        }
        dfs(u+1,sum+1,state);
    }else{
        dfs(u+1,sum,state);
        for (int i=0;i<ct;i++){
            if ((state>>i&1) && belong[u][i]){
                state^=1LL<<i;
            }
        }
        dfs(u+1,sum+1,state);
    }
}
int main(){
    freopen("matchstick.in","r",stdin);
    freopen("matchstick.out","w",stdout);
    int T;
    scanf("%d",&T);
    while (T--){
        int n,r;
        scanf("%d%d",&n,&r);
        m=2*n*(n+1);
        ct=0;
        memset(vis,0,sizeof(vis));
        memset(belong,0,sizeof(belong));
        for (int i=0,x;i<r;i++){
            scanf("%d",&x);
            vis[x]=true;
        }
        for (int sz=1;sz<=n;sz++){			// 正方形边长.cpp
            for (int i=0;i+sz-1<n;i++){		// 左上角的行数.cpp
                for (int j=0;j+sz-1<n;j++){ // 左上角的列数.cpp
                    int x0=i*(n+n+1)+1+j;	//上边.cpp
                    int x1=x0+sz*(n+n+1);	//下边.cpp
                    int x2=x0+n;			//左边.cpp
                    int x3=x2+sz;			//右边.cpp
                    bool ok=true;
                    for (int k=0;k<sz;k++){
                        if (vis[x0+k] || vis[x1+k] ||
                           vis[x2+(n+n+1)*k] || vis[x3+(n+n+1)*k]){
                            ok=false;
                            break;
                        }
                    }
                    /* 对于一个正方形来讲,那 4 个变量是这个正方形上下两条边中最左
                     边的那根棍,和左右两条边中最上边的那根棍。但是这个正方形肯定不
                     只这四根棍,这个正方形还有哪些棍呢,就是上下两条边往右画,
                     左右两条边往下画.cpp
                    */
                    if (ok){
                        mmax[ct]=x1+sz-1;//从x1开始sz个,最后一个是x1+sz-1.cpp
                        d[ct].clear();
                        for (int k=0;k<sz;k++){
                            belong[x0+k][ct]=true;
                            belong[x1+k][ct]=true;
                            belong[x2+(n+n+1)*k][ct]=true;
                            belong[x3+(n+n+1)*k][ct]=true;
                            d[ct].push_back(x0+k);
                            d[ct].push_back(x1+k);
                            d[ct].push_back(x2+(n+n+1)*k);
                            d[ct].push_back(x3+(n+n+1)*k);
                        }
                        ct++;
                    }
                }
            }
        }
        ans=1e6;
        dfs(1,0,(1LL<<ct)-1);
        printf("%d\n",ans);
    }
    return 0;
}
//好累啊

此题主要还靠你的毅力······我好累······
请给我点点赞或者说说我此博文的不足,谢谢!

猜你喜欢

转载自blog.csdn.net/Keven_11/article/details/105455014