hdu5925 Coconuts 【离散化+bfs】

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5;
struct node
{
int x,y;
}p[1024];
int x
,y
,g[2048][2048],d
,vis[2048][2048],mx,my;
int dx[]={0,0,-1,1};
int dy[]={1,-1,0,0};
ll vx
,vy
;
vector<ll>ans;

ll bfs(int sx,int sy)
{
ll sum=0;
queue<int>qx;
queue<int>qy;
while(!qx.empty()) qx.pop();
while(!qy.empty()) qy.pop();
qx.push(sx);
qy.push(sy);
vis[sx][sy]=1;
sum+=vx[sx]*vy[sy];
while(!qx.empty())
{
int tx=qx.front();
int ty=qy.front();
qx.pop();
qy.pop();
//printf("   %d %d %lld\n",tx,ty,sum);
for(int i=0;i<4;i++)
{
int nx=tx+dx[i];
int ny=ty+dy[i];
if(nx<1||nx>mx||ny<1||ny>my||g[nx][ny]||vis[nx][ny]) continue;
vis[nx][ny]=1;
qx.push(nx);
qy.push(ny);
sum+=vx[nx]*vy[ny];
}
}
return sum;
}
int main()
{
int _,n,m,k,a,b,t,__;
scanf("%d",&_);
for(int __=1;__<=_;__++)
{
scanf("%d%d%d",&n,&m,&k);
a=b=0;
x[a++]=0;
y[b++]=0;
x[a++]=n;
y[b++]=m;
for(int i=0;i<k;i++)
{
scanf("%d%d",&p[i].x,&p[i].y);
x[a++]=p[i].x;
x[a++]=p[i].x-1;
y[b++]=p[i].y;
y[b++]=p[i].y-1;
}
sort(x,x+a);
a=unique(x,x+a)-x;
t=1;
d[0]=x[0];
for(int i=1;i<a;i++)
{
if(x[i]!=x[i-1]+1)
d[t++]=x[i-1]+1;
d[t++]=x[i];
}
sort(d,d+t);
mx=t-1;
for(int i=1;i<t;i++)
{
vx[i]=d[i]-d[i-1];
}
for(int i=0;i<k;i++)
{
p[i].x=lower_bound(d,d+t,p[i].x)-d;
}

sort(y,y+b);
b=unique(y,y+b)-y;
t=1;
d[0]=y[0];
for(int i=1;i<b;i++)
{
if(y[i]!=y[i-1]+1)
d[t++]=y[i-1]+1;
d[t++]=y[i];
}
sort(d,d+t);
my=t-1;
for(int i=1;i<t;i++)
{
vy[i]=d[i]-d[i-1];
//printf("%d %d\n",i,vy[i]);
}
for(int i=0;i<k;i++)
{
p[i].y=lower_bound(d,d+t,p[i].y)-d;
}

memset(g,0,sizeof(g));
memset(vis,0,sizeof(vis));
for(int i=0;i<k;i++)
{
g[p[i].x][p[i].y]=1;
//printf("%d %d\n",p[i].x,p[i].y);
}
//printf("%d %d\n",mx,my);
ans.clear();
//printf("%d\n",vis[1][1]);
for(int i=1;i<=mx;i++)
{
for(int j=1;j<=my;j++)
{
if(vis[i][j]==0&&!g[i][j])
{
//printf("~~~%d %d\n",i,j);
ans.push_back(bfs(i,j));
}
}
}
printf("Case #%d:\n",__);
printf("%d\n",ans.size());
sort(ans.begin(),ans.end());
for(int i=0;i<ans.size();i++)
{
if(i) printf(" ");
printf("%lld",ans[i]);
}
printf("\n");
}
return 0;
}