HDU 3642 Get The Treasury ( 线段树 求长方体体积并 )

求覆盖三次及其以上的长方体体积并。

这题跟 http://wenku.baidu.com/view/d6f309eb81c758f5f61f6722.html 这里讲的长方体体积并并不一样。

因为本题Z坐标范围非常小，所以可以离散化Z坐标，枚举每个体积块。

对每一个体积块：用底面积＊高求其体积。底面积直接用“线段树求长方形面积并”来得到即可。

对于覆盖次数，pushUp的时候：

1.满足 当前覆盖次数大于等于3的，直接求线段长。

2.小于3的，由 左右儿子覆盖次数＝3 - 当前覆盖次数 的两个儿子更新上来得到。

#include <cstdio>

#include <cstdlib>

#include <cstring>

#include <algorithm>



using namespace std;



#define lson l, m, rt << 1

#define rson m + 1, r, rt << 1 | 1

#define lc rt << 1

#define rc rt << 1 | 1

#define LL long long int



const int MAXN = 2010;



struct cube

{

    int x1, y1, z1;

    int x2, y2, z2;

    void readCube()

    {

        scanf( "%d%d%d", &x1, &y1, &z1 );

        scanf( "%d%d%d", &x2, &y2, &z2 );

        return;

    }

}cb[MAXN];



struct Line

{

    int s;  //sao ru sao chu

    int x;

    int y1, y2;

    Line() {}

    Line( int s, int x, int y1, int y2 ): s(s), x(x), y1(y1), y2(y2) { }

    void showLine()

    {

        printf("s=%d x=%d y1=%d y2=%d\n", s, x, y1, y2 );

        return;

    }

};



struct node

{

    int cnt;

    int len[3];

};



int N;

node Tr[ MAXN << 3 ];

Line L[ MAXN << 3 ];

int Z[MAXN << 1];

int Y[MAXN << 2]; //li san hua

int cntZ, cntY;



bool cmp( const Line& a, const Line &b )

{

    if ( a.x != b.x ) return a.x < b.x;

    return a.s > b.s;

}



void build( int l, int r, int rt )

{

    for ( int i = 0; i < 3; ++i )

        Tr[rt].len[i] = 0;

    Tr[rt].cnt = 0;

    if ( l == r ) return;

    int m = ( l + r ) >> 1;

    build( lson );

    build( rson );

    return;

}



void PushUp( int rt, int l, int r )

{

    //len0

    if ( Tr[rt].cnt > 0 )

        Tr[rt].len[0] = Y[r+1] - Y[l];

    else

        Tr[rt].len[0] = Tr[lc].len[0] + Tr[rc].len[0];



    //len1

    if ( Tr[rt].cnt > 1 )

        Tr[rt].len[1] = Y[r+1] - Y[l];

    else if ( Tr[rt].cnt == 1 )

        Tr[rt].len[1] = Tr[lc].len[0] + Tr[rc].len[0];

    else

        Tr[rt].len[1] = Tr[lc].len[1] + Tr[rc].len[1];



    //len2

    if ( Tr[rt].cnt > 2 )

        Tr[rt].len[2] = Y[r+1] - Y[l];

    else if ( Tr[rt].cnt == 2 )

        Tr[rt].len[2] = Tr[lc].len[0] + Tr[rc].len[0];

    else if ( Tr[rt].cnt == 1 )

        Tr[rt].len[2] = Tr[lc].len[1] + Tr[rc].len[1];

    else

        Tr[rt].len[2] = Tr[lc].len[2] + Tr[rc].len[2];



    return;

}



void Update( int L, int R, int v, int l, int r, int rt )

{

    if ( L <= l && r <= R )

    {

        Tr[rt].cnt += v;

        PushUp( rt, l, r );

        return;

    }

    if ( l == r ) return;



    int m = ( l + r ) >> 1;



    if ( L <= m ) Update( L, R, v, lson );

    if ( R > m )  Update( L, R, v, rson );



    PushUp( rt, l, r );

    return;

}



int main()

{

    int T, cas = 0;

    scanf( "%d", &T );

    while ( T-- )

    {

        scanf( "%d", &N );

        cntZ = 0;

        cntY = 0;

        for ( int i = 0; i < N; ++i )

        {

            cb[i].readCube();

            Z[cntZ++] = cb[i].z1;

            Z[cntZ++] = cb[i].z2;

            Y[cntY++] = cb[i].y1;

            Y[cntY++] = cb[i].y2;

        }



        sort( Z, Z + cntZ );

        sort( Y, Y + cntY );

        cntZ = unique( Z, Z + cntZ ) - Z;

        cntY = unique( Y, Y + cntY ) - Y;



        LL ans = 0;

        for ( int i = 0; i < cntZ - 1; ++i )

        {

            int cntL = 0;

            for ( int j = 0; j < N; ++j )

            {

                if ( cb[j].z1 <= Z[i] && cb[j].z2 >= Z[i + 1] )

                {

                    L[cntL++] = Line( 1, cb[j].x1, cb[j].y1, cb[j].y2 );

                    L[cntL++] = Line(-1, cb[j].x2, cb[j].y1, cb[j].y2 );

                }

            }



            sort( L, L + cntL, cmp );

            build( 0, cntY - 1, 1 );



            for ( int k = 0; k < cntL; ++k )

            {

                if ( k )

                    ans += (LL)( L[k].x-L[k-1].x )*Tr[1].len[2]*( Z[i+1]-Z[i] );

                int a = lower_bound( Y, Y + cntY, L[k].y1 ) - Y;

                int b = lower_bound( Y, Y + cntY, L[k].y2 ) - Y - 1;   //这里一定要减1！！！！！！

                Update( a, b, L[k].s, 0, cntY - 1, 1 );

            }

        }



        printf( "Case %d: %lld\n", ++cas, ans );

    }

    return 0;

}

之前一直调试不出结果，是因为点树和线段树没分清楚，就是线段树节点中存的是一个点，还是一个单位长度倍数的线段。

对于二分得到a，b值，b值需要－1，而pushUp的时候r值需要＋1这里并不是很理解。