线段树扫描线（体积并）hdu3642

对z离散化，然后遍历一遍z，每次求面积并，然后底面积乘上z长

刚开始忘用long long 了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
const int maxn=2000;


struct Rect
{
    int x1,y1,z1,x2,y2,z2;
}rect[maxn];

struct seg
{
    int l,r,h,c;
    seg(){}
    seg(int a,int b,int d,int t):l(a),r(b),h(d),c(t){}
    bool operator<(const seg &a)const
    {
        if(h==a.h)return c>a.c;
        return h<a.h;
    }
}s[maxn<<1];

int n,cntx,cntz;
int x[maxn],z[maxn];


struct IntervalTree
{
    int sum[maxn<<3],one[maxn<<3],twice[maxn<<3],cnt[maxn<<3];
    void build()
    {
        memset(sum,0,sizeof(sum));
        memset(one,0,sizeof(one));
        memset(twice,0,sizeof(twice));
        memset(cnt,0,sizeof(cnt));
    }
    void maintain(int o,int l,int r)
    {
        if(cnt[o]==3)
        {
            sum[o]=x[r+1]-x[l];
            one[o]=0,twice[o]=0;
        }
        else if(cnt[o]==2)
        {
            sum[o]=one[o<<1]+one[o<<1|1]+twice[o<<1]+twice[o<<1|1]+sum[o<<1]+sum[o<<1|1];
            one[o]=0;
            twice[o]=x[r+1]-x[l]-sum[o];//覆盖两次以上的减去覆盖三次以上的就覆盖改两次的
        }
        else if(cnt[o]==1)
        {
            sum[o]=twice[o<<1]+twice[o<<1|1]+sum[o<<1]+sum[o<<1|1];
            twice[o]=one[o<<1]+one[o<<1|1];
            one[o]=x[r+1]-x[l]-sum[o]-twice[o];
        }
        else if(cnt[o]==0)
        {
            sum[o]=sum[o<<1]+sum[o<<1|1];
            one[o]=one[o<<1]+one[o<<1|1];
            twice[o]=twice[o<<1]+twice[o<<1|1];
        }
    }
    void update(int o,int l,int r,int q1,int q2,int t)
    {
        if(q1<=l&&r<=q2)
        {
            cnt[o]+=t;
            maintain(o,l,r);
            return;
        }
        int mid=(l+r)>>1;
        if(q1<=mid)update(o<<1,l,mid,q1,q2,t);
        if(q2>mid)update(o<<1|1,mid+1,r,q1,q2,t);
        maintain(o,l,r);
    }
};


void solve()
{
    sort(x,x+cntx);
    sort(z,z+cntz);
    cntz=unique(z,z+cntz)-z;
    cntx=unique(x,x+cntx)-x;
    long long ans=0;
    for(int i=0;i<cntz-1;i++)
    {
        int num=0;
        for(int j=0;j<n;j++)
        {
            if(rect[j].z1<=z[i]&&rect[j].z2>z[i])
            {
                int x1=rect[j].x1,x2=rect[j].x2;
                int y1=rect[j].y1,y2=rect[j].y2;
                s[num++]=seg(x1,x2,y1,1);
                s[num++]=seg(x1,x2,y2,-1);
            }

        }
        sort(s,s+num);
        long long area=0;
        IntervalTree tree;
        tree.build();
        for(int j=0;j<num-1;j++)
        {
            int l=lower_bound(x,x+cntx,s[j].l)-x;
            int r=lower_bound(x,x+cntx,s[j].r)-x-1;
            tree.update(1,0,cntx-1,l,r,s[j].c);
            area+=(long long)tree.sum[1]*(s[j+1].h-s[j].h);
        }
        ans+=area*(z[i+1]-z[i]);
    }
    cout<<ans<<endl;

}

int main()
{
    freopen("in.txt","r",stdin);
    int T,cas=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        cntx=cntz=0;
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d%d%d%d",&rect[i].x1,&rect[i].y1,&rect[i].z1,&rect[i].x2,&rect[i].y2,&rect[i].z2);
            x[cntx++]=rect[i].x1;x[cntx++]=rect[i].x2;
            z[cntz++]=rect[i].z1;z[cntz++]=rect[i].z2;

        }
        printf("Case %d: ",cas++);
        if(n<3){printf("0\n");continue;}
        solve();
    }
    return 0;
}