[CF 316F3]Suns and Rays解题报告

作者注

图片来自

http://www.cnblogs.com/wangck/p/4294282.html

题目大意

给出一张位图，是太阳和周围的光芒，像这样：

问有多少个太阳，并统计每个太阳周围的光线数量，排序后输出。

图的规模是1600*1600，光线的宽度<=3.保证太阳不会太小，光线也不会太短。

题解

这题挺有意思。

首先将白色“收缩”：如果一个点的周围有黑色，就将它变成黑的。重复若干次。

就变成了这样。

然后“扩张”，即逆操作：若周围有白的就把它变成白的。

（注意每个白色区域都变大了）

我们可以在这张图中找出所有太阳的“核心部分”，并给它们标号。

用原图减掉这张图，得到：

这样就把所有的光线独立了出来。对它们进行floodfill即可。

注意，最后可能有一些小块，判断一下，如果规模太小就不认为是光线即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int DIR_NUM=4;
const int SIZEH=1610,SIZEC=100010;
int H,W;
int C;//核心数量
int board[SIZEH][SIZEH];
int core[SIZEH][SIZEH];
int sun[SIZEH][SIZEH];
int raynum[SIZEC]={0};
int dx[]={0,0,1,-1,1,1,-1,-1},dy[]={1,-1,0,0,1,-1,1,-1};
void print(int s[SIZEH][SIZEH]){
	for(int i=0;i<H;i++){
		for(int j=0;j<W;j++){
			if(s[i][j]==-1) printf("-");
			else printf("%d",s[i][j]);
		}
		printf("\n");
	}
}
int nowtot=0;
int DFS_ray(int x,int y){
	sun[x][y]=-1;
	nowtot++;
	int ans=0;
	for(int d=0;d<DIR_NUM;d++){
		int x1=x+dx[d],y1=y+dy[d];
		if(0<=x1&&x1<H&&0<=y1&&y1<W){
			if(sun[x1][y1]>0) ans=sun[x1][y1];
			else if(sun[x1][y1]==0){
				int tp=DFS_ray(x1,y1);
				if(tp) ans=tp;
			}
		}
	}
	return ans;
}
void Find_Ray(void){
	for(int i=0;i<H;i++){
		for(int j=0;j<W;j++){
			if(sun[i][j]==0){
				nowtot=0;
				int t=DFS_ray(i,j);
				if(nowtot>=5) raynum[t]++;
			}
		}
	}
}
void DFS_core(int x,int y,int col){
	core[x][y]=col;
	for(int d=0;d<DIR_NUM;d++){
		int x1=x+dx[d],y1=y+dy[d];
		if(0<=x1&&x1<H&&0<=y1&&y1<W){
			if(core[x1][y1]==0) DFS_core(x1,y1,col);
		}
	}
}
void Find_Core(void){
	int timer=0;
	for(int i=0;i<H;i++){
		for(int j=0;j<W;j++){
			if(core[i][j]==0){
				DFS_core(i,j,++timer);
			}
		}
	}
	C=timer;
}
void Background_Spread(int f[SIZEH][SIZEH],int g[SIZEH][SIZEH]){//p向外扩散
	for(int i=0;i<H;i++){
		for(int j=0;j<W;j++){
			g[i][j]=f[i][j];
			for(int d=0;d<DIR_NUM;d++){
				int i1=i+dx[d],j1=j+dy[d];
				if(0<=i1&&i1<H&&0<=j1&&j1<W){
					if(f[i1][j1]==-1) g[i][j]=-1;
				}
			}
		}
	}
}
void Multiple_Background_Spread(int f[SIZEH][SIZEH],int tim){
	static int g[SIZEH][SIZEH];
	for(int i=1;i<=tim;i++){
		Background_Spread(f,g);
		memcpy(f,g,sizeof(g));
	}
}
void Sun_Recover(int f[SIZEH][SIZEH],int g[SIZEH][SIZEH]){
	for(int i=0;i<H;i++){
		for(int j=0;j<W;j++){
			g[i][j]=max(f[i][j],board[i][j]);
			for(int d=0;d<DIR_NUM;d++){
				int i1=i+dx[d],j1=j+dy[d];
				if(0<=i1&&i1<H&&0<=j1&&j1<W){
					if(board[i][j]>=0&&f[i1][j1]>0) g[i][j]=f[i1][j1];
				}
			}
		}
	}
}
void Multiple_Sun_Recover(int f[SIZEH][SIZEH],int tim){
	static int g[SIZEH][SIZEH];
	for(int i=1;i<=tim;i++){
		Sun_Recover(f,g);
		memcpy(f,g,sizeof(g));
	}
}
void Work(void){
	int width=3;//三次就能消去所有光束
	memcpy(core,board,sizeof(core));
	Multiple_Background_Spread(core,width);
	Find_Core();
	memcpy(sun,core,sizeof(sun));
	Multiple_Sun_Recover(sun,width);
	Find_Ray();
	sort(raynum+1,raynum+1+C);
	printf("%d\n",C);
	for(int i=1;i<=C;i++) printf("%d ",raynum[i]);
	printf("\n");
}
void Read(void){
	scanf("%d%d",&H,&W);
	for(int i=0;i<H;i++){
		for(int j=0;j<W;j++){
			scanf("%d",&board[i][j]);
			board[i][j]--;
		}
	}
}
int main(){
	//freopen("t1.in","r",stdin);
	//freopen("t1.out","w",stdout);
	Read();
	Work();
	return 0;
}