HDU-3642 Get The Treasury

线段树+扫描线+离散化

对立方体进行扫描，求出覆盖三次及以上的体积

可以对Z轴离散化，然后在普通的扫描线算法外再套一层对Z轴的循环

每次求出的二维面积乘以Z轴高度加起来就是要求的体积

扫描线时用三个数组保存覆盖了一次，两次，三次的线段长度，还有cover数组保存区间覆盖次数

更新函数详见代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

using namespace std;
typedef long long ll;
const int maxn=2300+10; //这里要开大一些，因为后面更新函数没有设置限制，会查找溢出的子节点位置

struct line
{
    int x,y1,y2,z1,z2;
    int f;
    bool operator < (const line &u) const
    {
        return x<u.x;
    }
}edge[maxn],e[maxn];
int len1[maxn<<2],len2[maxn<<2],len3[maxn<<2];
int cover[maxn<<2];
int Y[maxn],Z[maxn];

void PushUp(int rt,int l,int r)
{
    if(cover[rt]>=3)
    {
        len3[rt]=Y[r]-Y[l];
        len1[rt]=len2[rt]=0;
    }
    else if(cover[rt]==2)
    {
        len3[rt]=len3[rt<<1]+len3[rt<<1|1]+len1[rt<<1]+len1[rt<<1|1]+len2[rt<<1]+len2[rt<<1|1];
        len2[rt]=Y[r]-Y[l]-len3[rt];
        len1[rt]=0;
    }
    else if(cover[rt]==1)
    {
        len3[rt]=len3[rt<<1]+len3[rt<<1|1]+len2[rt<<1]+len2[rt<<1|1];
        len2[rt]=len1[rt<<1]+len1[rt<<1|1];
        len1[rt]=Y[r]-Y[l]-len3[rt]-len2[rt];
    }
    else
    {
        len3[rt]=len3[rt<<1]+len3[rt<<1|1];
        len1[rt]=len1[rt<<1]+len1[rt<<1|1];
        len2[rt]=len2[rt<<1]+len2[rt<<1|1];
    }
}

void update(int L,int R,int f,int l,int r,int rt)
{
    if(Y[l]>=L&&Y[r]<=R)
    {
        cover[rt]+=f;
    }
    else
    {
        if(r-l==1) return;
        int m=(l+r+1)>>1;
        if(L<Y[m]) update(L,R,f,l,m,rt<<1);
        if(R>Y[m]) update(L,R,f,m,r,rt<<1|1);
    }
    PushUp(rt,l,r);
}

int main()
{
    //freopen("/home/zlwang/test.txt","r",stdin);
    int n,T,kase=0;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        int x1,x2,y1,y2,z1,z2;
        int cnt=0;
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d%d%d%d",&x1,&y1,&z1,&x2,&y2,&z2);
            Y[i]=y1;
            Y[n+i]=y2;
            Z[i]=z1;
            Z[n+i]=z2;
            edge[cnt].y1=y1;
            edge[cnt].y2=y2;
            edge[cnt].z1=z1;
            edge[cnt].z2=z2;
            edge[cnt].x=x1;
            edge[cnt].f=1;
            cnt++;
            edge[cnt].y1=y1;
            edge[cnt].y2=y2;
            edge[cnt].z1=z1;
            edge[cnt].z2=z2;
            edge[cnt].x=x2;
            edge[cnt].f=-1;
            cnt++;
        }
        sort(edge,edge+2*n);
        sort(Y,Y+2*n);
        sort(Z,Z+2*n);
        int cnty=1,cntz=1;
        for(int i=1;i<2*n;i++)
        if(Y[i]!=Y[i-1]) Y[cnty++]=Y[i];
        for(int i=1;i<2*n;i++)
        if(Z[i]!=Z[i-1]) Z[cntz++]=Z[i];
        ll ans=0;
        for(int k=0;k<cntz-1;k++)
        {
            ll t=0;
            memset(cover,0,sizeof(cover));
            memset(len1,0,sizeof(len1));
            memset(len2,0,sizeof(len2));
            memset(len3,0,sizeof(len3));
            int cnte=0;
            for(int i=0;i<2*n;i++)
            if(edge[i].z1<=Z[k]&&edge[i].z2>=Z[k+1])
            e[cnte++]=edge[i];
            for(int i=0;i<cnte-1;i++)
            {
                update(e[i].y1,e[i].y2,e[i].f,0,cnty-1,1);
                t+=(ll)(len3[1]*1ll*(e[i+1].x-e[i].x));
            }
            ans+=t*(Z[k+1]-Z[k]);
        }
        printf("Case %d: %lld\n",++kase,ans);
    }
    return 0;
}