传送门:http://codeforces.com/contest/616/problem/C
比赛的时候,只想把每个位置所在连通块的数量都记录下来,
其实分着记录一下就好了,即记录每个位置的连通块号,再记录每个联通块号所对应的数量就可以了,这样就不会很麻烦了。
比赛时候记录的时候用memset T 了,改成单个修改后过了,实际用set就可以了
PS:改过后的代码中犯了2个错误,1个是输出的时候没有将字符串首字符加1,导致无输出,然后%10的时候最后一步忘记了,导致WRONG ON TEST 9,太马虎了!
再标记一下调编译类型的方法 GUNC++11:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e3+10;
char ans[maxn][maxn],s[maxn][maxn];
int vis[maxn][maxn];
int belong[maxn][maxn],size[(int)1e6+10];
int n,m;
int dir[4][2]={1,0,-1,0,0,1,0,-1};int sum,cur;
bool inside(int x,int y){
if(x>=1&&x<=n&&y>=1&&y<=m) return 1;
return 0;
}
void dfs(int x,int y){
vis[x][y]=1;
belong[x][y]=cur;size[cur]++;
for(int i=0;i<4;i++){
int xx=x+dir[i][0];
int yy=y+dir[i][1];
if((!inside(xx,yy))||(vis[xx][yy])||s[xx][yy]=='*') continue;
dfs(xx,yy);
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%s",s[i]+1);
}
cur=1;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(s[i][j]=='.'&&(!vis[i][j])){
dfs(i,j);cur++;
}
}
}
/* for(int i=1;i<cur;i++){
cout<<"i="<<i<<" "<<size[i]<<endl;
}
*/
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(s[i][j]=='.') ans[i][j]='.';
else{
set<int>f;
f.insert(belong[i+1][j]);
f.insert(belong[i-1][j]);
f.insert(belong[i][j+1]);
f.insert(belong[i][j-1]);
int sum=1;
for(int x:f){
sum+=(size[x])%10;
}
ans[i][j]=sum%10+'0';
}
//printf("i=%d j=%d ans=%c\n",i,j,ans[i][j]);
}
ans[i][m+1]='\0';
}
for(int i=1;i<=n;i++){
printf("%s\n",ans[i]+1);
}
return 0;
}
再附上比赛时的丑陋代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<math.h>
#include<stdlib.h>
#include<vector>
#include<map>
#include<string>
using namespace std;
#define ll long long
#define For(i,n) for(int i=1;i<=n;i++)
#define Dor(i,n) for(int i=n;i>=1;i--)
#define s(n) scanf("%d",&n)
#define p(n) printf("%d\n",n)
//#define maxn 200000+10
#define ms(n) memset(n,0,sizeof(n))
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
const double eps=1e-8;
const int inf=0x3f3f3f3f;
const int maxn=1e6+10;
char a[1005][1005];
int n,m;
int dir[4][2]={1,0,-1,0,0,1,0,-1};
int b[1005][1005];
int vis[1005][1005];
struct node{
int x,y;
};
int vis4[1000010];
vector<node>ans;
int count1,count2;
char c[1005][1005];
int d[1005][1005];
bool inside(int x,int y){
if(x>=1&&x<=n&&y>=1&&y<=m) return 1;
return 0;
}
int sum;node tmp;
void dfs(int x,int y){
int xx,yy;
tmp.x=x;tmp.y=y;
sum=(sum+1)%10;
ans.push_back(tmp);
vis[x][y]=1;
for(int i=0;i<4;i++){
xx=x+dir[i][0];
yy=y+dir[i][1];
if(vis[xx][yy]) continue;
if(inside(xx,yy)==0||a[xx][yy]=='*') continue;
dfs(xx,yy);
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%s",a[i]+1);
}
int count=1;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(!vis[i][j]&&a[i][j]=='.'){
ans.clear();
sum=0;
dfs(i,j);
for(int k=0;k<ans.size();k++){
b[ans[k].x][ans[k].y]=sum%10;
d[ans[k].x][ans[k].y]=count;
}
count++;
}
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
if(a[i][j]=='.'){
c[i][j]='.';
}
else{
c[i][j]=1;
//memset(vis4,0,sizeof(vis4));
ans.clear();
for(int k=0;k<4;k++){
int xx=i+dir[k][0];
int yy=j+dir[k][1];
if(!inside(xx,yy)) continue;
if(!vis4[d[xx][yy]]){
tmp.x=xx;tmp.y=yy;
ans.push_back(tmp);
vis4[d[xx][yy]]=1;
c[i][j]=(c[i][j]+b[xx][yy])%10;
}
}
for(int kk=0;kk<ans.size();kk++){
vis4[d[ans[kk].x][ans[kk].y]]=0;
}
c[i][j]+='0';
}
}
c[i][m+1]='\0';
}
/* for(int i=1;i<=n;i++){
for(int j=1;j<m;j++){
cout<<b[i][j]<<" ";
}
cout<<b[i][m]<<endl;
}
cout<<endl;*/
for(int i=1;i<=n;i++){
printf("%s\n",c[i]+1);
}
}