优雅的代码

@羊驼

#include<bits/stdc++.h>
using namespace std;
int n,sum;
char s[1005][1005];
int vis[1005][1005];
int dx[]={1,-1,0,0};
int dy[]={0,0,1,-1};
map<int,int> mp; 
void dfs1(int x,int y){
	vis[x][y]=sum;
	for(int i=0;i<4;i++){
		int xx=x+dx[i];
		int yy=y+dy[i];
		if(xx>=1 && xx<=n && yy>=1 && yy<=n && s[xx][yy]==s[x][y] && vis[xx][yy]==0){
			dfs1(xx,yy);
		}
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>s[i]+1;
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(s[i][j]!='.' && vis[i][j]==0){
				sum++;
				dfs1(i,j);
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			if(s[i][j]=='.'){
				set<int> se;
				for(int k=0;k<4;k++){
					int xx=i+dx[k];
					int yy=j+dy[k];
					if(xx>=1 && xx<=n && yy>=1 && yy<=n && vis[xx][yy])se.insert(vis[xx][yy]);
				}
				for(auto it:se){
					mp[it]++;
				}
			}
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cout<<mp[vis[i][j]]<<" ";
		}
		cout<<endl;
	}
	return 0;
}