给定一个 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
使用二进制进行状态压缩,枚举第一行所有翻转方法,共2^15种,然后从第二行开始遍历,如果该点上一行为1则翻转该点。最后判断矩阵是否全为0
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int N = 20;
int g[N][N],n,m,u,bg[N][N],flag=0;
int dx[5]={0,0,0,1,-1},dy[5]={1,-1,0,0,0};
inline void cz(int x,int y){
for(int i=0;i<5;i++){
int a=x+dx[i],b=dy[i]+y;
if(a>=0&&b>=0&&a<n&&b<m)bg[a][b]^=1;
}
}
bool ans[N][N],bans[N][N];//翻转矩阵
int f=1e9;
inline bool check(){
for(int i(0);i<n;i++)
for(int j(0);j<m;j++)
if(bg[i][j]==1)return 0;
return 1;
}
int main(){
scanf("%d%d",&n,&m);
for(int i(0);i<n;i++)
for(int j(0);j<m;j++)
scanf("%d",&g[i][j]);
for(int i=0;i<=(1<<m)-1;i++)
{ int res=0;
int k=i;
memcpy(bg,g,sizeof g);
memset(bans,0,sizeof bans);
for(int j=0;j<m;j++)
{
if(k>>j&1){ //为1就翻转
cz(0,j);
res++;
bans[0][j]=1;
}
}
for(int j=1;j<n;j++)
{
for(int s=0;s<m;s++){
if(bg[j-1][s]==1){
cz(j,s);
bans[j][s]=1;
res++;
}
}
}
if(check()&&f>res){
memcpy(ans,bans,sizeof bans);
flag=1;
f=res;
}
}
if(!flag)puts("IMPOSSIBLE");
else {
for(int i(0);i<n;i++)
{
for(int j(0);j<m;j++)
{
if(j)printf(" ");
printf("%d",ans[i][j]);
}
printf("\n");
}
}
}