JZOJ4419. 【GDOI2016模拟4.2】hole

题目大意

输入$n$个操作，每个操作$x,y,d$，当$d=0$时表示在坐标为$(x,y)$的地方打一个洞，$d>0$时表示查询以$(x,y) , (x+d,y) , (x,y+d)$三点围成的等腰直角三角形内（包括边界）有多少个洞。

30%的数据$n≤3333$.
另30% 的数据 只会在打完洞后才开始询问,$xi,yi≤333333$.
100%的数据 $1≤n≤88888，xi,yi≤3333333$ .
（第一次见这么奇怪的数据范围= =）

题解

30%的数据就不说了。

解法一

先考虑60%的数据。如下图。

所以我们可以排完序以后线段树维护。
100%的数据，既会有挖洞也会询问，那么怎么处理顺序呢？
你听说过安利（划掉）CDQ分治吗？
这里我们可以用CDQ分治的思想。先将操作序列分成两半，那么左边区间的挖洞操作会对右区间的询问造成影响。所以我们把左区间的挖洞和右区间的查询单独取出来，做上面60%的算法，累计答案，再迭代下去，最终的解。
时间复杂度$O(nlog_2^2n)$

解法二

直接上树套树。具体细节留给读者思考。

SRC

//解法一
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std ;

#define N 100000 + 10
#define M 5000000 + 10
struct Qtype {
    int x , y , d , h ;
} Q[N] , t[N] ;

int Son[4*M][2] , tot ;
int T[4*M] , ans[N] ;
int n , MAXX , MAXY , MAXXY , ret ;

bool cmp1( Qtype a , Qtype b ) { return a.x + a.y + a.d < b.x + b.y + b.d ? 1 : ( a.x + a.y + a.d == b.x + b.y + b.d && a.d < b.d ? 1 : 0 ) ; }
bool cmp2( Qtype a , Qtype b ) { return a.x < b.x ? 1 : ( a.x == b.x && a.d > b.d ? 1 : 0 ) ; }
bool cmp3( Qtype a , Qtype b ) { return a.y < b.y ? 1 : ( a.y == b.y && a.d > b.d ? 1 : 0 ) ; }

void ADD( int v , int l , int r , int x ) {
    if ( l == x && r == x ) {
        T[v] ++ ;
        return ;
    }
    int mid = (l + r) / 2 ;
    if ( x <= mid ) {
        if ( !Son[v][0] ) Son[v][0] = ++ tot , T[tot] = Son[tot][0] = Son[tot][1] = 0 ;
        ADD( Son[v][0] , l , mid , x ) ;
    }
    else {
        if ( !Son[v][1] ) Son[v][1] = ++ tot , T[tot] = Son[tot][0] = Son[tot][1] = 0 ;
        ADD( Son[v][1] , mid + 1 , r , x ) ;
    }
    T[v] = T[Son[v][0]] + T[Son[v][1]] ;
}

void Search( int v , int l , int r , int x , int y ) {
    if ( !v ) return ;
    if ( l == x && r == y ) {
        ret += T[v] ;
        return ;
    }
    int mid = (l + r) / 2 ;
    if ( y <= mid ) Search( Son[v][0] , l , mid , x , y ) ;
    else if ( x > mid ) Search( Son[v][1] , mid + 1 , r , x , y ) ;
    else {
        Search( Son[v][0] , l , mid , x , mid ) ;
        Search( Son[v][1] , mid + 1 , r , mid + 1 , y ) ;
    }
}

void CDQ_DIV( int l , int r ) {
    if ( l == r ) return ;
    int num = 0 , mid = (l + r) / 2 ;
    for (int i = l ; i <= r ; i ++ ) {
        if ( i <= mid && Q[i].d ) continue ;
        if ( i > mid && !Q[i].d ) continue ;
        t[++num] = Q[i] ;
    }
    int total = 0 ;
    sort( t + 1 , t + num + 1 , cmp1 ) ;
    for (int i = 1 ; i <= num ; i ++ ) {
        if ( !t[i].d ) total ++ ;
        else ans[t[i].h] += total ;
    }
    sort( t + 1 , t + num + 1 , cmp2 ) ;
    tot = 1 ;
    Son[1][0] = Son[1][1] = T[1] = 0 ;
    for (int i = 1 ; i <= num ; i ++ ) {
        if ( t[i].d ) {
            ret = 0 ;
            Search( 1 , 0 , MAXXY , 0 , t[i].x + t[i].y + t[i].d ) ;
            ans[t[i].h] -= ret ;
        } else ADD( 1 , 0 , MAXXY , t[i].x + t[i].y ) ;
    }
    tot = 1 ;
    Son[1][0] = Son[1][1] = T[1] = 0 ;
    for (int i = 1 ; i <= num ; i ++ ) {
        if ( t[i].d ) {
            ret = 0 ;
            Search( 1 , 0 , MAXY , 0 , t[i].y - 1 ) ;
            ans[t[i].h] += ret ;
        } else ADD( 1 , 0 , MAXY , t[i].y ) ;
    }
    sort( t + 1 , t + num + 1 , cmp3 ) ;
    tot = 1 ;
    Son[1][0] = Son[1][1] = T[1] = 0 ;
    for (int i = 1 ; i <= num ; i ++ ) {
        if ( t[i].d ) {
            ret = 0 ;
            Search( 1 , 0 , MAXXY , 0 , t[i].x + t[i].y + t[i].d ) ;
            ans[t[i].h] -= ret ;
        } else ADD( 1 , 0 , MAXXY , t[i].x + t[i].y ) ;
    }
    CDQ_DIV( l , mid ) ;
    CDQ_DIV( mid + 1 , r ) ;
}

int main() {
    freopen( "hole.in" , "r" , stdin ) ;
    freopen( "hole.out" , "w" , stdout ) ;
    scanf( "%d" , &n ) ;
    for (int i = 1 ; i <= n ; i ++ ) {
        scanf( "%d%d%d" , &Q[i].x , &Q[i].y , &Q[i].d ) ;
        MAXX = max( MAXX , Q[i].x ) ;
        MAXY = max( MAXY , Q[i].y ) ;
        MAXXY = max( MAXXY , Q[i].x + Q[i].y + Q[i].d ) ;
        Q[i].h = i ;
    }
    CDQ_DIV( 1 , n ) ;
    for (int i = 1 ; i <= n ; i ++ ) {
        if ( Q[i].d ) printf( "%d\n" , ans[i] ) ;
    }
    return 0 ;
}

以上.