【kuangbin】简单搜索 - 4.翻转【递归搜索+开关灯模型】

Date：2022.02.02
题意：
给定一个 M×N 的 01 矩阵。
你需要选择其中一些元素，并对选择的元素进行翻转操作。
翻转操作是指将所选元素以及与其上下左右相邻的元素（如果有）进行翻转（0 变 1，1 变 0）。
我们希望最终矩阵变为一个全 0 矩阵，并且选择进行翻转操作的元素数量尽可能少。
输出最佳翻转方案。
输入格式
第一行包含整数 M,N。
接下来 M 行，每行包含 N 个整数，每个整数为 0 或 1。
输出格式
共 M 行，每行包含 N 个整数，其中第 i 行第 j 列的整数，表示第 i 行第 j 列元素的翻转次数。
如果翻转操作次数最少的操作方案不唯一，则输出字典序最小的方案。
如果不存在合理方案，则输出 IMPOSSIBLE。
数据范围
1≤M,N≤15
输入样例：
4 4
1 0 0 1
0 1 1 0
0 1 1 0
1 0 0 1
输出样例：
0 0 0 0
1 0 0 1
1 0 0 1
0 0 0 0

思路：同此题。此外让操作数组字典序最小即让对第1行的操作尽可能小。
代码如下：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 20;
typedef long long LL;
char g[N][N],c[N][N],z[N][N];
LL n,m,k,t;
LL dx[5]={0,-1,1,0,0},dy[5]={0,0,0,1,-1};
void turn(LL x,LL y)
{
    for(int i=0;i<5;i++)
    {
        LL ix=x+dx[i],iy=y+dy[i];
        if(ix<0||iy<0||ix>=n||iy>=m) continue;
        g[ix][iy]^=1;
    }
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++) {cin>>g[i][j];c[i][j]='0';}
    LL ans=1e18;bool flag=false;
    for(int k=0;k<(1<<m);k++)
    {
        LL res=0;
        char backup[N][N];
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++) c[i][j]='0';
        memcpy(backup,g,sizeof g);
        for(int j=0;j<m;j++)
            if(k>>j&1) 
            {res++;turn(0,j);c[0][j]^=1;}
        for(int i=0;i<n-1;i++)
            for(int j=0;j<m;j++)
                if(g[i][j]=='1') 
                {res++;turn(i+1,j);c[i+1][j]^=1;}
        bool flag1=true;
        for(int j=0;j<m;j++)
            if(g[n-1][j]=='1') 
            {flag1=false;break;}
        if(flag1)
        {
            flag=true;
            if(ans>res) 
            {
                ans=res;
                memcpy(z,c,sizeof c);
            }
        }
        memcpy(g,backup,sizeof backup);
    }
    if(flag)
    {
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++) cout<<z[i][j]<<' ';
            cout<<endl;
        }
    }
    else cout<<"IMPOSSIBLE"<<endl;
    return 0;
}