【HDU3642 Get The Treasury】线段树之扫描线（体积并）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3642

题目大意：给你n个立方体，求相交区域大于等于三次的体积和。

题目大意：

    写这题之前我先做的hdu3255，有思路但是Wrong Answer到死。被逼无奈搜题解，都说是线段树扫描线求体积并，这之前我只会二维面积并。

    其实吧，三维的和二维的其实差不多。如果一个立方体的高为h，那么我们可以把它分割成h层，对每一层进行面积并的扫描，注意是从下往上。

   这题离散化x坐标是为了方便建树，离散化z坐标是为了节约时间。

   剩下的问题就变成了如何求覆盖大于等于三次体积范围。

   问题同样可以转化为二维的面积并求解，对每一层进行扫描，每层求覆盖大于等于三次的面积区域。每层求的结果加起来就是答案了。

  

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define lz 2*u,l,mid
#define rz 2*u+1,mid+1,r
const int maxn=2222;
typedef long long lld;
int flag[4*maxn];
lld sum1[4*maxn], sum2[4*maxn], sum3[4*maxn];
int X[maxn], Z[maxn];

struct Node
{
    int lx, rx, y, z1, z2, s;
    Node() {}
    Node(int lx_, int rx_ , int y_, int zm_, int zl_, int s_)
    {
        lx=lx_, rx=rx_, y=y_, z1=zm_, z2=zl_, s=s_;
    }
    bool operator<(const Node &S) const
    {
        if(y==S.y) return s>S.s;
        else return y<S.y;
    }
} line[maxn], tmp[maxn];

void push_up(int u, int l, int r)
{
    if(flag[u]>=3)   ///开始写成了flag[u]==3， wrong answer了一个晚上
    {
        sum3[u]=sum2[u]=sum1[u]=X[r+1]-X[l];
    }
    else if(flag[u]==2)
    {

        sum2[u]=sum1[u]=X[r+1]-X[l];
        if(l==r)sum3[u]=0;
        else
            sum3[u]=sum1[2*u]+sum1[2*u+1];
    }
    else if(flag[u]==1)
    {
        sum1[u]=X[r+1]-X[l];
        if(l==r)sum2[u]=sum3[u]=0;
        else
        {
            sum2[u]=sum1[2*u]+sum1[2*u+1];
            sum3[u]=sum2[2*u]+sum2[2*u+1];
        }
    }
    else
    {
        if(l==r)sum1[u]=sum2[u]=sum3[u]=0;
        else
        {
            sum1[u]=sum1[2*u]+sum1[2*u+1];
            sum2[u]=sum2[2*u]+sum2[2*u+1];
            sum3[u]=sum3[2*u]+sum3[2*u+1];
        }
    }
}

void Update(int u, int l, int r, int tl, int tr, int c)
{
    if(tl>tr) return ;
    if(tl<=l&&r<=tr)
    {
        flag[u]+=c;
        push_up(u,l,r);
        return ;
    }
    int mid=(l+r)>>1;
    if(tr<=mid) Update(lz,tl,tr,c);
    else if(tl>mid) Update(rz,tl,tr,c);
    else
    {
        Update(lz,tl,mid,c);
        Update(rz,mid+1,tr,c);
    }
    push_up(u,l,r);
}

int find(int tmp, int n)
{
    int l=1, r=n, mid;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(X[mid]==tmp) return mid;
        else if(X[mid]<tmp) l=mid+1;
        else r=mid-1;
    }
}

int main()
{
    int T, n, tcase=0;
    cin >> T;
    while(T--)
    {
        cin >> n ;
        int num=0;
        for(int i=0; i<n; i++)
        {
            int x1, y1, x2, y2, z1, z2;
            scanf("%d%d%d%d%d%d",&x1,&y1,&z1,&x2,&y2,&z2);
            line[++num]=Node(x1,x2,y1,z1,z2,1);
            X[num]=x1, Z[num]=z1;
            line[++num]=Node(x1,x2,y2,z1,z2,-1);
            X[num]=x2, Z[num]=z2;
        }
        sort(line+1,line+num+1);
        sort(X+1,X+num+1);
        sort(Z+1,Z+num+1);
        int ep=1, m=1;
        for(int i=2; i<=num; i++)
            if(X[i]!=X[ep]) X[++ep]=X[i];
        for(int i=2; i<=num; i++)
            if(Z[i]!=Z[m]) Z[++m]=Z[i];
        lld ans=0;
        for(int j=1; j<m; j++)
        {
            memset(flag,0,sizeof(flag));
            memset(sum1,0,sizeof(sum1));
            memset(sum2,0,sizeof(sum2));
            memset(sum3,0,sizeof(sum3));
            lld tp=0, cnt=0;
            for(int i=1; i<=num; i++)
            {
                if(line[i].z1<=Z[j]&&Z[j]<line[i].z2)
                    tmp[++cnt]=line[i];
            }
            for(int i=1; i<cnt; i++)
            {
                int l=find(tmp[i].lx,ep);
                int r=find(tmp[i].rx,ep)-1;
                Update(1,1,ep-1,l,r,tmp[i].s);
                tp+=(lld)sum3[1]*(lld)(tmp[i+1].y-tmp[i].y);
            }
            ans+=(lld)tp*(lld)(Z[j+1]-Z[j]);
        }
        printf("Case %d: %I64d\n",++tcase,ans);
    }
    return 0;
}