思路:按题意模拟,因为提前消除会对后面的判断有影响,所以多开一个数组用来消除,用原数组作比较。(一开始打算dfs,后面发现没必要)
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 35;
int a[N][N],b[N][N];
int nex[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
int n,m;
void dfs(int x,int y,int flag)
{
b[x][y]=0;
if(flag==0) //消除列
{
for(int i=x+1;i<n;i++)
{
if(a[i][y]==a[x][y])
b[i][y]=0;
else
break;
}
for(int i=x-1;i>=0;i++)
{
if(a[i][y]==a[x][y])
b[i][y]=0;
else
break;
}
}
else//消除行
{
for(int j=y+1;j<m;j++)
{
if(a[x][j]==a[x][y])
b[x][j]=0;
else
break;
}
for(int j=y-1;j>=0;j--)
{
if(a[x][j]==a[x][y])
b[x][j]=0;
else
break;
}
}
}
int main()
{
cin>>n>>m;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
cin>>a[i][j],b[i][j]=a[i][j];
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
if(i+2<n&&a[i][j]==a[i+1][j]&&a[i][j]==a[i+2][j])
dfs(i,j,0);
if(j+2<m&&a[i][j]==a[i][j+1]&&a[i][j]==a[i][j+2])
dfs(i,j,1);
}
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
cout<<b[i][j]<<" ";
cout<<endl;
}
return 0;
}