基本思路:要将被1包围的0改为2,很自然想到用DFS或者BFS算法,这里我使用DFS算法。然而,要找到被1包围的0并不容易,我们可以将问题转化为将不被1包围的0改为2,被1包围的0不变,而只需在输出时做判断修改输出既可。
#include<iostream>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<math.h>
using namespace std;
int n;
int a[32][32];
int dx[4]={1,-1,0,0};
int dy[4]={0,0,1,-1};
void dfs(int x,int y)
{
if(a[x][y]==1||a[x][y]==2)
{
return;
}
a[x][y]=2;
for(int i=0;i<4;i++)
{
if(x+dx[i]>=0&&x+dx[i]<n&&y+dy[i]>=0&&y+dy[i]<n)
{
dfs(x+dx[i],y+dy[i]);
}
}
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
cin>>a[i][j];
}
}
for(int i=0;i<n;i++)
{
dfs(i,0);
dfs(0,i);
dfs(n-1,i);
dfs(i,n-1);
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
if(a[i][j]==2)
{
cout<<"0 ";
}
if(a[i][j]==1)
{
cout<<"1 ";
}
if(a[i][j]==0)
{
cout<<"2 ";
}
}
cout<<"\n";
}
return 0;
}