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[N],y[N],g[2048][2048],d[N],vis[2048][2048],mx,my;
int dx[]={0,0,-1,1};
int dy[]={1,-1,0,0};
ll vx[N],vy[N];
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;
}