打标记的DFS
题目链接:https://codeforces.com/contest/616/problem/C
题意:
给你n行m列的矩阵,矩阵*表示障碍,.表示空地
对于每一个障碍,让你输出去掉这个障碍之后(此点算作空点),这个点所在的连通块的大小是多少
注:每个障碍点单独处理且答案需要%10
思路:
一开始想到直接BFS,但是这样应该会T,所以就想能不能换一种方法。
正解:
求出每个联通块的方格的个数,这样在求障碍物对应的联通块大小是就可以就可以直接计算它四周的联通块大小即可。
注意:
1、对每一个障碍物而言,来自四个相邻格子的联通块可能是同一个,这时可以用一个 f [ ][ ] 数组记录这是第几个联通块以去重
2、每次dfs搜索时把空点放入队列中并随时更新已经搜索的点数(最后即为联通块大小),此次dfs完成后将此联通块大小赋给队列中的每个点。
AC代码:
#include<iostream>
#include<cstdio>
#include<set>
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
#include<cmath>
#include<string>
#include<cstring>
#include<cstdlib>
#include<map>
using namespace std;
#define scan(n) scanf("%d",&n)
#define ll long long
struct ss
{
int r,c;
};
typedef struct ss ss;
queue<ss> q;
int a[1005][1005];
int num[1005][1005];
int r[1005][1005];
int f[1005][1005];
int cnt;
int n,m;
int tot=1;
void dfs(int i,int j)
{
if(i<1||i>n||j<1||j>m)
return;
if(num[i][j])
return;
if(a[i][j]==1)
return;
ss s;
s.r=i;
s.c=j;
q.push(s);
cnt++;
num[i][j]=1;
dfs(i+1,j);
dfs(i-1,j);
dfs(i,j+1);
dfs(i,j-1);
return;
}
int main()
{
int i,j;
char ch;
cin>>n>>m;
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
{
cin>>ch;
if(ch=='*')
a[i][j]=1;
else
a[i][j]=0;
}
memset(num,0,sizeof(num));
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
if(a[i][j]==0&&num[i][j]==0)
{
cnt=0;
dfs(i,j);
while(!q.empty())
{
ss s=q.front();
q.pop();
int rr=s.r;
int cc=s.c;
num[rr][cc]=cnt;
f[rr][cc]=tot;
}
tot++;
}
}
}
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
r[i][j]=num[i-1][j];
int f1=f[i-1][j],f2=f[i+1][j],f3=f[i][j-1],f4=f[i][j+1];
if(f2!=f1)
r[i][j]+=num[i+1][j];
if(f3!=f1&&f3!=f2)
r[i][j]+=num[i][j-1];
if(f4!=f1&&f4!=f2&&f4!=f3)
r[i][j]+=num[i][j+1];
r[i][j]++;
r[i][j]%=10;
}
}
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
if(a[i][j]==1)
cout<<r[i][j];
else
cout<<'.';
}
cout<<endl;
}
return 0;
}