Problem 1581 - Union of cubes+武大校赛+线段树维护

题目 http://acm.whu.edu.cn/land/problem/detail?problem_id=1581

题意：在一个三维空间里，有N*(K*N)*(K*K*N)个点，每个点有一定的值，然后有M个长方体，每个长方体以固定位置（（x1,y1,z1）（x2,y2,z2）为对角位置，其中(x1<=x2)(y1<=y2)(z1<=z2)）按顺序放在这个三维空间里，可以有重叠部分，每次计算长方体的联合体所占的三维空间里的点的值得总和，要求最后把最大的中间值输出。这题一看就是个模拟，可是如果简单的暴力的话绝对会T，时间复杂度比较高。可以看到，X轴的范围是1到10，Y的是1到100，Z的是1到1000，我们何不建立X*Y棵线段树，然后对Z轴这个线段进行维护，每次取出z1到z2范围内的数值之后，就把其清0，表示该数值以取完，这里不就是区间更新和区间查询了，所以用一个线段树来维护一下就行了。(比赛的时候写出来了，可是以前不怎么写线段树，忘了rt应该要是从1开始的，样例都没过掉==)

code:

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int Sum[11][101][1001*4];
bool lazy[11][101][1001*4];
const int inf=0x3f3f3f3f;
void build(int x,int y,int l,int r,int rt)
{
    if (l==r)
    {
        scanf("%d",&Sum[x][y][rt]);
    }
    else
    {
        int m=(l+r)/2;
        build(x,y,l,m,rt*2);
        build(x,y,m+1,r,rt*2+1);
        Sum[x][y][rt]=Sum[x][y][rt*2]+Sum[x][y][rt*2+1];
    }
}
void push_down(int x,int y,int rt)
{
    if (lazy[x][y][rt])
    {
        Sum[x][y][rt]=0;
        Sum[x][y][rt*2]=0;
        Sum[x][y][rt*2+1]=0;
        lazy[x][y][rt*2]=lazy[x][y][rt];
        lazy[x][y][rt*2+1]=lazy[x][y][rt];
        lazy[x][y][rt]=false;
    }
}
void push_up(int x,int y,int rt)
{
    Sum[x][y][rt]=Sum[x][y][rt*2+1]+Sum[x][y][rt*2];
}
void update(int x,int y,int l,int r,int rt,int sl,int sr)
{
    if (sl<=l&&sr>=r)
    {
        lazy[x][y][rt]=true;
        Sum[x][y][rt]=0;
    }
    else
    {
        push_down(x,y,rt);
        int m=(l+r)/2;
        if (m>=sl)
        {
            update(x,y,l,m,rt*2,sl,sr);
        }
        if (m<sr)
        {
            update(x,y,m+1,r,rt*2+1,sl,sr);
        }
        push_up(x,y,rt);
    }
}
int query(int x,int y,int rt,int l,int r,int ql,int qr)
{
    if (lazy[x][y][rt])
    {
        return 0;
    }
    else if (ql<=l&&qr>=r)
    {
        return Sum[x][y][rt];
    }
    else
    {
        int m=(l+r)/2;
        int ans=0;
        if (m>=ql)
        {
            ans+=query(x,y,rt*2,l,m,ql,qr);
        }
        if (m<qr)
        {
            ans+=query(x,y,rt*2+1,m+1,r,ql,qr);
        }
        return ans;
    }
}
int main()
{
    freopen("in.txt","r",stdin);
    int N,K,M;
    while (~scanf("%d%d%d",&N,&K,&M))
    {
        memset(Sum,0,sizeof(Sum));
        memset(lazy,false,sizeof(lazy));
        for (int i=1;i<=N;i++)
        {
            for (int j=1;j<=K*N;j++)
            {
                build(i,j,1,K*K*N,1);
            }
        }
        int sum_=0,Max=-inf;
        for (int i=0;i<M;i++)
        {
            int x1,y1,z1,x2,y2,z2;
            scanf("%d%d%d%d%d%d",&x1,&y1,&z1,&x2,&y2,&z2);
            for (int j=x1;j<=x2;j++)
            {
                for (int k=y1;k<=y2;k++)
                {
                    sum_+=query(j,k,1,1,K*K*N,z1,z2);
                    update(j,k,1,K*K*N,1,z1,z2);
                }
            }
            Max=max(Max,sum_);
        }
        printf("%d\n",Max);
    }
    return 0;
}