Get The Treasury HDU - 3642(线段树求矩形面积交加强)

求立体几何的交体积，与二维求矩形面积交相同，写个模板

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

using namespace std;

#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define ll long long

const int MAXN=4444;

struct Line{
    int l,r,h,s;
    Line(int _l,int _r,int _h,int _s)
    {
        l=_l;
        r=_r;
        s=_s;
        h=_h;
    }
    bool operator <(const struct Line &b)const{
        return h<b.h;
    }
};

struct edge{
    int x1,y1,z1;
    int x2,y2,z2;
    edge(int _x1,int _y1,int _z1,int _x2,int _y2,int _z2){
        x1=_x1;
        y1=_y1;
        z1=_z1;
        x2=_x2;
        y2=_y2;
        z2=_z2;
    }
};

vector<int> vt1;
vector<Line> vt2;
vector<edge> vt3;
vector<int> vt4;

int sum[MAXN<<2];
int sum2[MAXN<<2];
int sum3[MAXN<<2];
int lazy[MAXN<<2];

void pushup(int l,int r,int rt)
{
    if(lazy[rt]) sum[rt]=vt4[r+1]-vt4[l];
    else if(l==r) sum[rt]=0;
    else sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    
    if(lazy[rt]>=2) sum2[rt]=vt4[r+1]-vt4[l];
    else if(l==r) sum2[rt]=0;
    else if(lazy[rt]==1) sum2[rt]=sum[rt<<1]+sum[rt<<1|1];
    else sum2[rt]=sum2[rt<<1]+sum2[rt<<1|1];
    
    if(lazy[rt]>=3) sum3[rt]=vt4[r+1]-vt4[l];
    else if(l==r) sum3[rt]=0;
    else if(lazy[rt]==2) sum3[rt]=sum[rt<<1]+sum[rt<<1|1];
    else if(lazy[rt]==1) sum3[rt]=sum2[rt<<1]+sum2[rt<<1|1];
    else sum3[rt]=sum3[rt<<1]+sum3[rt<<1|1];
}

// void pushup(int left, int right, int root) {
//     if(lazy[root] >= 3) {
//         sum3[root] = sum2[root] = sum[root] = vt4[right+1] - vt4[left];
//     } else if(lazy[root] == 2) {
//         sum2[root] = sum[root] = vt4[right+1] - vt4[left];
//         sum3[root] = sum[root << 1] + sum[root << 1 | 1];
//     } else if(lazy[root] == 1) {
//         sum[root] = vt4[right+1] - vt4[left];
//         sum3[root] = sum2[root << 1] + sum2[root << 1 | 1];
//         sum2[root] = sum[root << 1] + sum[root << 1 | 1];
//     } else {
//         sum3[root] = sum3[root << 1] + sum3[root << 1 | 1];
//         sum2[root] = sum2[root << 1] + sum2[root << 1 | 1];
//         sum[root] = sum[root << 1] + sum[root << 1 | 1];
//     }
// }
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        lazy[rt]+=c;
        pushup(l,r,rt);
        return;
    }
    int mid=(l+r)>>1;
    if(L<=mid) update(L,R,c,lson);
    if(R>mid) update(L,R,c,rson);
    pushup(l,r,rt);
}
int main()
{
    int n;
    scanf("%d",&n);
    int kace=1;
    while(n--)
    {
        int m;
        scanf("%d",&m);
        vt1.clear();
        vt3.clear();
        for(int i=0;i<m;i++)
        {
            int x,y,z,x1,y1,z1;
            scanf("%d%d%d%d%d%d",&x,&y,&z,&x1,&y1,&z1);
            vt1.push_back(z);
            vt1.push_back(z1);
            vt3.push_back(edge(x,y,z,x1,y1,z1));
        }
        sort(vt1.begin(),vt1.end());
        vt1.erase(unique(vt1.begin(),vt1.end()),vt1.end());
        ll ans=0;
        for(int i=0;i<vt1.size()-1;i++)
        {
            vt2.clear();
            vt4.clear();
            memset(sum,0,sizeof(sum));
            memset(sum2,0,sizeof(sum2));
            memset(sum3,0,sizeof(sum3));
            memset(lazy,0,sizeof(lazy));
            for(int j=0;j<vt3.size();j++)
            {
                if(vt3[j].z1<=vt1[i]&&vt1[i+1]<=vt3[j].z2)
                {
                    vt2.push_back(Line(vt3[j].x1,vt3[j].x2,vt3[j].y1,1));
                    vt2.push_back(Line(vt3[j].x1,vt3[j].x2,vt3[j].y2,-1));
                    vt4.push_back(vt3[j].x1);
                    vt4.push_back(vt3[j].x2);
                }
            }
            sort(vt2.begin(),vt2.end());
            sort(vt4.begin(),vt4.end());
            vt4.erase(unique(vt4.begin(),vt4.end()),vt4.end());
            for(int z=0;z<vt2.size()-1;z++)
            {
                int l=lower_bound(vt4.begin(),vt4.end(),vt2[z].l)-vt4.begin();
                int r=lower_bound(vt4.begin(),vt4.end(),vt2[z].r)-vt4.begin()-1;
                update(l,r,vt2[z].s,0,vt4.size()-1,1);
                ans+=1ll*sum3[1]*(vt2[z+1].h-vt2[z].h)*(vt1[i+1]-vt1[i]);
            }
        }
        printf("Case %d: %lld\n",kace++,ans);
    }
    return 0;
}