HDU 4031 Attack （线段树）

成功袭击次数=所有袭击次数-成功防守次数

需要一个辅助pre来记录上一次袭击成功什么时候，对于每个查询，从上一次袭击成功开始，每隔t更新一次。

感觉这样做最坏时间复杂度是O(n^2)，这里 说是O(q)*O(q/t)log(n)，应该是超时边缘。

能AC或许因为数据不强。

PS.初始化的时候直接memset会超时，每次根据N的大小build线段树，memset也根据N的大小来就能AC。

 

#include <cstdio>

#include <cstring>

#include <cstdlib>



#define lson l, m, rt << 1

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

#define lc rt << 1

#define rc rt << 1 | 1



using namespace std;



const int MAXN = 20100;



struct node

{

    int l, r;

};



node D[MAXN];

int sum[ MAXN << 2 ];

int flag[ MAXN << 2 ];

int pre[ MAXN ];

int defen[MAXN];

int N, Q, t;



void PushDown( int rt, int m )

{

    if ( flag[rt] )

    {

        flag[lc] += flag[rt];

        flag[rc] += flag[rt];

        sum[lc] += flag[rt]*(m-(m>>1));

        sum[rc] += flag[rt]*(m>>1);

        flag[rt] = 0;

    }

    return;

}



void update( int L, int R, int l, int r, int rt )

{

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

    {

        sum[rt] += 1;

        flag[rt] += 1;

        return;

    }

    PushDown(rt, r - l + 1);

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

    if ( L <= m ) update( L, R, lson );

    if ( R > m )  update( L, R, rson );

    return;

}



int query( int L, int l, int r, int rt )

{

    if ( L == l && r == L ) return sum[rt];

    PushDown(rt, r - l + 1);

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

    if ( L <= m ) return query( L, lson );

    else return query( L, rson );

}



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

{

    flag[rt] = sum[rt] = 0;

    if ( l == r ) return;

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

    build( lson );

    build( rson );

    return;

}



int main()

{

    int T, cas = 0;

    scanf( "%d", &T );

    char str[10];

    while( T-- )

    {

        scanf( "%d%d%d", &N, &Q, &t );

        build( 1, N, 1 );

        memset( defen, 0, sizeof(int)*(N+4) );

        memset( pre, 0, sizeof(int)*(N+4) );

        memset( D, 0, sizeof(D[0])*(N+4) );



        int cnt = 0;

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

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

        {

            scanf( "%s", str );

            if ( str[0] == 'A' )

            {

                scanf("%d%d", &D[cnt].l, &D[cnt].r );

                update( D[cnt].l, D[cnt].r, 1, N, 1 );

                ++cnt;

            }

            else

            {

                int a;

                scanf( "%d", &a );

                for ( int i = pre[a]; i <= cnt; )

                {

                    if ( a >= D[i].l && a <= D[i].r )

                    {

                        pre[a] = i + t;

                        ++defen[a];

                        i += t;

                    }

                    else ++i;

                }

                //printf("qry=%d def=%d\n", query(a, 1, N, 1 ), defen[a] );

                printf( "%d\n", query(a, 1, N, 1 ) - defen[a] );

            }

            //printf("***qry=%d def=%d\n", query(5, 1, N, 1 ), defen[5] );

        }

    }

    return 0;

}