寒假算法训练4-B（从四条边的每一个点进行dfs）

Sample Input
1
5 5
100 253 214 146 120
123 0 0 0 0
54 0 33 47 0
255 0 0 78 0
14 11 0 0 0
Sample Output
0 0 0 0 0
0 0 0 0 0
0 0 33 47 0
0 0 0 78 0
0 0 0 0 0

#include <bits/stdc++.h>
using namespace std;
const int maxw=1500,maxh=1500;
int dis[4][2]={1,0,-1,0,0,1,0,-1};
int h,w,t;
int a[maxh][maxw];

void dfs(int x,int y)
{
    a[x][y]=0;
    for(int i=0;i<4;i++)
    {
        int tx=x+dis[i][0];
        int ty=y+dis[i][1];
        if(tx>=0 && ty>=0 && tx<h && ty<w && a[tx][ty]!=0) dfs(tx,ty);
    }
}
int main()
{
    cin>>t;
    while(t--)
    {
        memset(a,-1,sizeof(a));
        cin>>w>>h;
        for(int i=0;i<h;i++)
            for(int j=0;j<w;j++) cin>>a[i][j];
        for(int i=0;i<w;i++)
        {
            if(a[0][i]!=0) dfs(0,i);
            if(a[0][h-1]!=0) dfs(0,h-1);
        }
        for(int i=0;i<h;i++)
        {
            if(a[i][0]!=0) dfs(i,0);
            if(a[i][w-1]!=0) dfs(i,w-1);
        }
        for(int i=0;i<h;i++)
        {
            for(int j=0;j<w;j++) cout<<a[i][j]<<" ";
            cout<<endl;
        }
    }
    return 0;
}