POJ 1222 EXTENDED LIGHTS OUT【暴力dfs】

最新推荐文章于 2020-09-01 01:32:44 发布

原创最新推荐文章于 2020-09-01 01:32:44 发布 · 402 阅读

0 ·

CC 4.0 BY-SA版权

深搜-dfs 专栏收录该内容

26 篇文章

订阅专栏

本文探讨了一道关于5*6矩阵的操作题，通过状态压缩的方法，利用DFS实现求解将初始矩阵所有元素变为0的操作序列。文章提供了一个具体的算法实现思路及代码示例。

点击打开链接

题意：

给你一个5*6 的方格，每个方格里面分别由1or0.

每次操作一个方格，会把该方格and该方格的上下左右四个方向的方格里面的数字取反，

现在给你一个已经有数字的方格，让你给出一个能把方格全部变成0的方案，

题解：

很明显的状态压缩，枚举第一行后面的都可以递推出来，2^6*24的复杂度。

我直接dfs了第一行。

//#include <bits/stdc++.h>
#include<stdio.h>
#include<iostream>
#include<stdlib.h>
#include<string.h>
#define ll long long
using namespace std;
const int maxn=2e5+10;
int a[10][10],ans[10][10];
int c[10],b[10][10],f;
void dfs(int pos){
    if(f) return ;
    if(pos==7){
        f=1;
        memset(ans,0,sizeof(ans));
        for(int i=1;i<=5;++i)for(int j=1;j<=6;++j) b[i][j]=a[i][j];
        for(int i=1;i<=6;++i){
            ans[1][i]=c[i];
            if(c[i]==1){
                b[1][i]=!b[1][i];
                b[2][i]=!b[2][i];
                b[1][i-1]=!b[1][i-1];
                b[1][i+1]=!b[1][i+1];
            }
        }
        for(int i=2;i<=5;++i){
            for(int j=1;j<=6;++j){
                if(b[i-1][j]==1){
                    b[i-1][j]=!b[i-1][j];
                    b[i+1][j]=!b[i+1][j];
                    b[i][j]=!b[i][j];
                    b[i][j-1]=!b[i][j-1];
                    b[i][j+1]=!b[i][j+1];
                    ans[i][j]=1;
                }
            }
        }
        for(int i=1;i<=6;++i)
            if(b[5][i]==1){
                f=0;return ;
            }
        if(f) return ;
    }
    if(pos>=7) return;
    for(int i=1;i<=6;++i){
        c[pos]=1;
        dfs(pos+1);if(f) return ;
        c[pos]=0;
        dfs(pos+1);if(f) return ;
    }
    return ;
}
int main(){
    int T,ca=0;
    scanf("%d",&T);
    while(T--){
        f=0;
        for(int i=1;i<=5;++i)
            for(int j=1;j<=6;++j)
                scanf("%d",&a[i][j]);
        dfs(1);
        printf("PUZZLE #%d\n",++ca);
        for(int i=1;i<=5;++i){
            printf("%d",ans[i][1]);
            for(int j=2;j<=6;++j){
                printf(" %d",ans[i][j]);
            }puts("");
        }
    }
    return 0;
}