BZOJ-2120: 数颜色(BIT套SBT)

题目:http://www.lydsy.com/JudgeOnline/problem.php?id=2120

写裸数据结构题最开心啦~~~

维护一下与每个位置颜色一样的前面和后面最近的数,不存在则设为无穷,然后每次查询就等价于查询该区间后缀大于r的数有多少个,然后就各种BST暴力维护就可以啦。

代码(差点就超时了。。。):

d788d43f8794a4c2817d01480cf41bd5ad6e3973.jpg.png
#include <cstdio>

#include <algorithm>

#include <cstring>

#include <set>

 

using namespace std ;

 

#define MAXN 10010

#define inf 0x7fffffff

#define MAXC 1000100

#define MAXV 500010

#define L( t ) t -> left

#define R( t ) t -> right

#define K( t ) t -> key

#define S( t ) t -> size

#define update( t )S( t )=S(L( t ))+S(R( t ))+1

#define lowbit( t )((- t )& t )

 

struct SBT{

     

    struct node{

        node *left ,*right ;

        int key , size ;

    }*roof ,*null ;

     

    SBT(  ){

        roof = null =new( node );

        S( null )=0,L( null )=R( null )= null ;

    }

     

    void left( node*&t ){

        node *k =R( t );

        R( t )=L( k );update( t );

        L( k )= t ;update( k );

        t = k ;

    }

     

    void right( node*&t ){

        node *k =L( t );

        L( t )=R( k );update( t );

        R( k )= t ;update( k );

        t = k ;

    }

     

    void maintain( node*&t ){

        if(S(L(L( t )))>S(R( t ))){

            right( t );

            maintain(R( t ));maintain( t );

            return;

        }

        if(S(R(L( t )))>S(R( t ))){

            left(L( t ));right( t );

            maintain(L( t )),maintain(R( t ));maintain( t );

            return;

        }

        if(S(R(R( t )))>S(L( t ))){

            left( t );

            maintain(L( t ));maintain( t );

            return;

        }

        if(S(L(R( t )))>S(L( t ))){

            right(R( t ));left( t );

            maintain(L( t )),maintain(R( t ));maintain( t );

            return;

        }

    }

     

    void Insert(int k , node*&t ){

        if( t == null ){

            t =new( node );

            L( t )=R( t )= null ,S( t )=1,K( t )= k ;

            return;

        }

        Insert( k , k <K( t )?L( t ):R( t ));

        update( t );maintain( t );

    }

     

    void Delete(int k , node*&t ){

        if( k ==K( t )){

            if(L( t )== null ){

                node *u = t ;

                t =R( t );

                delete( u );

                return;

            }else if(R( t )== null ){

                node *u = t ;

                t =L( t );

                delete( u );

                return;

            }else{

                right( t );

                Delete( k ,R( t ));

            }

        }else Delete( k , k <K( t )?L( t ):R( t ));

        update( t );maintain( t );

    }

      

    int rank(int k , node *t ){

        if( t == null )return 0;

        return k <K( t )?rank( k ,L( t ))+S(R( t ))+1:rank( k ,R( t ));

    }

     

    void Push(int k ){

        Insert( k , roof );

    }

     

    void Erase(int k ){

        Delete( k , roof );

    }

     

    int Rank(int k ){

        return rank( k , roof );

    }

     

} sbt[ MAXN ];

 

set <int> rbt[ MAXC ];

 

int n , m , a[ MAXN ], suff[ MAXN ];

 

void Insert_bit(int pos ,int k ){

    for(int i = pos ; i <= n ; i +=lowbit( i )) sbt[ i ].Push( k );

}

 

void Delete_bit(int pos ,int k ){

    for(int i = pos ; i <= n ; i +=lowbit( i )) sbt[ i ].Erase( k );

}

 

int Rank_bit(int pos ,int k ){

    int rec =0;

    for(int i = pos ; i ; i -=lowbit( i )) rec += sbt[ i ].Rank( k );

    return rec ;

}

 

int main(  ){

    scanf("%d%d",&n ,&m );

    for(int i =0; i ++< n ;)scanf("%d",&a[ i ]);

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

        rbt[ i ].clear(  );

        rbt[ i ].insert(- inf ), rbt[ i ].insert( inf );

    }

    for(int i =0; i ++< n ;) rbt[ a[ i ]].insert( i );

    for(int i =0; i ++< n ;) suff[ i ]=*(++ rbt[ a[ i ]].lower_bound( i ));

    for(int i =0; i ++< n ;)Insert_bit( i , suff[ i ]);

    while( m --){

        int ch , l , r ;for( ch =getchar(  );!( ch =='Q'|| ch =='R'); ch =getchar(  ));

        if( ch =='Q'){

            scanf("%d%d",&l ,&r );

            printf("%d\n",Rank_bit( r , r )-Rank_bit( l -1, r ));

        }else{

            scanf("%d%d",&l ,&r );

            int k =*(-- rbt[ a[ l ]].lower_bound( l ));

            if( k >- inf ){

                Delete_bit( k , suff[ k ]);

                Insert_bit( k , suff[ k ]= suff[ l ]);

            }

            rbt[ a[ l ]].erase( l );

            Delete_bit( l , suff[ l ]);

            a[ l ]= r ;

            rbt[ r ].insert( l );

            set <int>:: iterator u = rbt[ r ].lower_bound( l );

            k =*(-- u );

            if( k >- inf ){

                Delete_bit( k , suff[ k ]);

                Insert_bit( k , suff[ k ]= l );

            }

            suff[ l ]=*(++ rbt[ r ].lower_bound( l ));

            Insert_bit( l , suff[ l ]);

        }

    }

    return 0;

}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,657评论 6 505
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,889评论 3 394
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,057评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,509评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,562评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,443评论 1 302
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,251评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,129评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,561评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,779评论 3 335
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,902评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,621评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,220评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,838评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,971评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,025评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,843评论 2 354

推荐阅读更多精彩内容