HDU 3255 Farming(线段树:扫描线)

HDU 3255 Farming(线段树:扫描线)

http://acm.hdu.edu.cn/showproblem.php?pid=3255

题意:多个可能重合的长方体求体积.

分析:

       该题只要把种子的价值转化为Z轴坐标[0,pi],就可以看成了重叠长方体就总体积的问题.解法类似HDU3642:

http://blog.csdn.net/u013480600/article/details/22625139

由于这里x与y的坐标范围都很大,所以离散化处理x坐标,然后扫描线以y为高.

       Z轴可以离散化,也可以不离散化,这里我们离散化.总体的流程就是:读取输入数据,把每个农田转化为一个立方体,从小到大枚举Z轴坐标,然后遍历一遍所有包含当前Z轴坐标的立方体生成对应的两条扫描线.然后根据这些扫描线求出Z[i]到Z[i+1]之间的矩形面积,乘以Z[i+1]-Z[i]就是这部分的体积了.

       线段树采用线信息保留型线段树,维护的信息有:

cnt:当前下位边-上位边数目的结果

sum:当前被覆盖的总长度.

只需要PushUp和build就行.

AC代码:1734ms,一次AC

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAXN=60000+1000;
#define lson i*2,l,m
#define rson i*2+1,m+1,r
struct seg
{
    int l,r,h,d;
    seg(){};
    seg(int l,int r,int h,int d):l(l),r(r),h(h),d(d){}
    bool operator <(const seg &b)const
    {
        return h<b.h;
    }
}ss[MAXN];
int cnt[MAXN*4],sum[MAXN*4];
int X[MAXN],p[10];
struct point
{
    int x,y,z;
    point(){}
    point(int x,int y,int z):x(x),y(y),z(z){}
};
struct cub
{
    point a,b;
}cube[MAXN/2];
void PushUp(int i,int l,int r)
{
    if(cnt[i])
    {
        sum[i]=X[r+1]-X[l];
    }
    else if(l==r)
        sum[i]=0;
    else
        sum[i]=sum[i*2]+sum[i*2+1];
}
void update(int ql,int qr,int v,int i,int l,int r)
{
    if(ql<=l && r<=qr)
    {
        cnt[i]+=v;
        PushUp(i,l,r);
        return ;
    }
    int m=(r+l)>>1;
    if(ql<=m) update(ql,qr,v,lson);
    if(m<qr) update(ql,qr,v,rson);
    PushUp(i,l,r);
}

int main()
{
    int T;
    scanf("%d",&T);
    for(int kase=1;kase<=T;kase++)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        p[0]=0;//表示Z轴的第0个不同的坐标
        for(int i=1;i<=m;i++)
            scanf("%d",&p[i]);
        int cntx=0;
        for(int i=1;i<=n;i++)
        {
            int x1,y1,x2,y2,s;
            scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&s);
            cube[i].a=point(x1,y1,0);
            cube[i].b=point(x2,y2,p[s]);
            X[cntx++]=x1;
            X[cntx++]=x2;
        }
        sort(p,p+m+1);
        sort(X,X+cntx);
        cntx=unique(X,X+cntx)-X;
        long long ans=0;
        for(int k=0;k<m;k++)
        {
            int cnt_ss=0;//扫描线的条数
            for(int i=1;i<=n;i++)if(cube[i].a.z<=p[k] && cube[i].b.z>p[k])
            {
                ss[cnt_ss++]=seg(cube[i].a.x,cube[i].b.x,cube[i].a.y,1);
                ss[cnt_ss++]=seg(cube[i].a.x,cube[i].b.x,cube[i].b.y,-1);
            }
            sort(ss,ss+cnt_ss);
            memset(cnt,0,sizeof(cnt));
            memset(sum,0,sizeof(sum));
            long long res=0;
            for(int i=0;i<cnt_ss-1;i++)
            {
                int ql=lower_bound(X,X+cntx,ss[i].l)-X;
                int qr=lower_bound(X,X+cntx,ss[i].r)-X-1;
                if(ql<=qr) update(ql,qr,ss[i].d,1,0,cntx-1);
                res += (long long)sum[1]*(ss[i+1].h-ss[i].h);//求面积
            }
            ans += res *(p[k+1]-p[k]);//求体积
        }
        printf("Case %d: %I64d\n",kase,ans);
    }
    return 0;
}