Educational Codeforces Round 5 C. The Labyrinth

题目链接~~

题意：求矩阵中’*’为阻塞，’.’为空，求每个阻塞块相邻的连通块有多少空块

如下：

input

3 3

* . *

. * .

* . *

output

3.3

.5.

3.3

由于每个’*’都搜一次会超时，所以现将每个连通块都搜出来，并给每个‘.’所在的块标上块号，并记录每个块的‘.’数量。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <map>
#include <set>
#include <string>
#include <vector>
using namespace std;

const int N=1005;
char mp

;
int cc

;
int num[N*N];
bool vis

;
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};
int n,m;
set<int>s;
set<int>::iterator it;
void dfs(int x,int y,int c){
cc[x][y]=c;
num[c]++;
for(int i=0;i<4;i++){
int xx=x+dx[i];
int yy=y+dy[i];
if(xx>=n||xx<0||yy<0||yy>=m)
continue;
if(cc[xx][yy]||mp[xx][yy]!='.') continue;
vis[xx][yy]=1;
dfs(xx,yy,c);
}
}
int work(int x,int y){
s.clear();
int ans=0;
for(int i=0;i<4;i++){
int xx=x+dx[i];
int yy=y+dy[i];
if(xx>=n||xx<0||yy<0||yy>=m)
continue;
if(mp[xx][yy]!='.') continue;
s.insert(cc[xx][yy]);
}
for(it=s.begin();it!=s.end();it++){
int t=*it;
ans+=num[t];
}
return ans;
}

int main()
{

scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
scanf("%s",mp[i]);
}
int c=1;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(cc[i][j]==0&&mp[i][j]=='.'){
num[c]=0;
dfs(i,j,c);
c++;
}
}
}
for(int i=0;i<n;i++)
for(int j=0;j<m;j++){
if(mp[i][j]=='*'){
int ans=(work(i,j)+1)%10;
mp[i][j]='0'+ans;
}
}
for(int i=0;i<n;i++)
printf("%s\n",mp[i]);

}