CodeForces 616 C. The Labyrinth(dfs)

Description
给出一个由’.’和’#’组成的n*m矩阵,问每个’#’周围’.’的个数(包括自己)
Input
第一行两个整数n和m表示矩阵行列数,之后为一n*m矩阵(1<=n,m<=1000)
Output
输出一个n*m矩阵,如果原矩阵对应位置是’.’则输出’.’,是’#’则输出这个’#’周围’.’的数量(包括自己),结果模10
Sample Input
这里写图片描述
Sample Output
46..3
..732
.6.4.
5.4.3
Solution
dfs,给每个连通块编号并记录其中’.’的数量,对于每个’#’,将其周围连通块的数量累计即可,注意一个’#’周围一个连通块可能会出现两次,但数值只能记录一次
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 1111
int n,m,res,cnt;
int dx[]={-1,0,1,0};
int dy[]={0,-1,0,1};
int mark[maxn][maxn],num[maxn*maxn],t[4];
char Map[maxn][maxn],ans[maxn][maxn];
void dfs(int x,int y)
{
    cnt++,ans[x][y]='.',Map[x][y]='#',mark[x][y]=res;
    for(int i=0;i<4;i++)
    {
        int xx=x+dx[i],yy=y+dy[i];
        if(xx<=0||xx>n||yy<=0||yy>m||Map[xx][yy]!='.')continue;
        dfs(xx,yy);
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(mark,0,sizeof(mark));
        num[0]=res=0;
        for(int i=1;i<=n;i++)scanf("%s",Map[i]+1);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                if(Map[i][j]=='.')
                    cnt=0,res++,dfs(i,j),num[res]=cnt;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                if(Map[i][j]=='*')
                {
                    int temp=1;
                    t[1]=num[mark[i][j-1]],t[2]=num[mark[i][j+1]],
                    t[3]=num[mark[i-1][j]],t[4]=num[mark[i+1][j]];
                    temp+=num[mark[i][j-1]],num[mark[i][j-1]]=0;
                    temp+=num[mark[i][j+1]],num[mark[i][j+1]]=0;
                    temp+=num[mark[i-1][j]],num[mark[i-1][j]]=0;
                    temp+=num[mark[i+1][j]],num[mark[i+1][j]]=0;
                    num[mark[i][j-1]]=t[1],num[mark[i][j+1]]=t[2],
                    num[mark[i-1][j]]=t[3],num[mark[i+1][j]]=t[4];
                    ans[i][j]=temp%10+'0';
                }
        for(int i=1;i<=n;i++)printf("%s\n",ans[i]+1);
    }
    return 0;
} 

你可能感兴趣的:(CodeForces 616 C. The Labyrinth(dfs))