首页 > 代码库 > [rope大法好] STL里面的可持久化平衡树--rope

[rope大法好] STL里面的可持久化平衡树--rope

简单用法:

#include <ext/rope>using namespace __gnu_cxx;int a[1000];rope<int> x;rope<int> x(a,a + n);rope<int> a(x);x->at(10);x[10];x->push_back(x)     // 在末尾添加xx->insert(pos,x)    // 在pos插入xx->erase(pos,x)     // 从pos开始删除x个x->replace(pos,x)   // 从pos开始换成xx->substr(pos,x)    // 提取pos开始x个

 

例题一:

IOI2012

scrivener

题意

设计支持如下 3 种操作:
1.T x:在文章末尾打下一个小写字母 x。(type 操作) 
2.U x:撤销最后的x 次修改操作。(Undo 操作)
(注意Query 操作并不算修改操作)
3.Q x:询问当前文章中第x 个字母并输出。(Query 操作)

操作数n<=100000 在线算法

clj都说这是道rope傻逼题……

 

#include<cstdio>#include<cstring>#include<cctype>#include<iostream>#include<algorithm>#include<ext/rope>using namespace std;using namespace __gnu_cxx;const int maxn=1e5+10;rope<char> *his[maxn];int n;int d[maxn];inline int lowbit(int x){ return x&-x; }inline void updata(int x){ while(x<=n){ d[x]++; x+=lowbit(x); } }inline int get(int x){ int res=0; while(x){ res+=d[x]; x-=lowbit(x); }return res; }inline char getC(){ char ch=getchar(); while(!isalpha(ch))ch=getchar(); return ch; }inline int getint(){    int res=0; char ch,ok=0;    while(ch=getchar())    {        if(isdigit(ch)){ res*=10;res+=ch-‘0‘;ok=1; }        else if(ok)break;    }    return res;}void deb(rope<char> s){    for(int i=0;i<s.length();i++)        cout<<s[i];puts("");}int main(){//    freopen("type.in","r",stdin);//    freopen("type.out","w",stdout);    n=getint();    his[0]=new rope<char>();    for(int i=1;i<=n;i++)    {        his[i]=new rope<char>(*his[i-1]);        // deb(*his[i]);        char opt=getC();        if(opt==‘T‘)        {            char x=getC();            his[i]->push_back(x);            updata(i);        }        else if(opt==‘U‘)        {            updata(i);            int x=getint();            int l=1,r=i,mid,now=get(i);            while(l<r)            {                mid=(l+r)>>1;                if(now-get(mid)>x) l=mid+1;                else r=mid;            }            his[i]=his[l-1];        }        else if(opt==‘Q‘)        {            int x=getint()-1;            putchar(his[i]->at(x));            putchar(‘\n‘);        } // deb(*his[i]);    }    return 0;}

 

其中,实现可持久化的操作是:his[i]=new rope<char>(*his[i-1]);

他可以实现O(1)的copy历史版本,因为rope的底层是红黑树,所以copy时只需copy根指针。

一键持久化……

 

例题二:

BZOJ 3673: 可持久化并查集 by zky

n个集合 m个操作
操作:
1 a b 合并a,b所在集合
2 k 回到第k次操作之后的状态(查询算作操作)
3 a b 询问a,b是否属于同一集合,是则输出1否则输出0

0<n,m<=2*10^4

分析:直接可持久化并查集的fa数组就可以了。

ACCode:

#include <cstdio>#include <ext/rope>using namespace std;using namespace __gnu_cxx;const int maxm = 20010;rope<int> *rp[maxm];int find(int i,int x){    if(rp[i]->at(x) == x)   return x;    int f = find(i,rp[i]->at(x));    if(f == rp[i]->at(x))   return f;    rp[i]->replace(x,f);    return rp[i]->at(x);}inline void merge(int i,int x,int y){    x = find(i,x),y = find(i,y);    if(x != y)  rp[i]->replace(y,x);}int a[maxm],lastans;int main(){    int n,m;    scanf("%d%d",&n,&m);    for(int i = 1;i <= n;i ++)  a[i] = i;    rp[0] = new rope<int> (a,a + n + 1);    for(int i = 1;i <= m;i ++)    {        rp[i] = new rope<int> (*rp[i - 1]);        int opt;    scanf("%d",&opt);        if(opt == 1)        {            int a,b;    scanf("%d%d",&a,&b);            merge(i,a/* ^ lastans*/,b/* ^ lastans*/);        }        else if(opt == 2)        {            int k;  scanf("%d",&k);            rp[i] = rp[k/* ^ lastans*/];        }        else        {            int a,b;    scanf("%d%d",&a,&b);            printf("%d\n",lastans = (find(i,a/* ^ lastans*/) == find(i,b/* ^ lastans*/)));        }    }    return 0;}

 

例题三:

AHOI2006文本编辑器editor

题意

设计数据结构支持

插入删除反转字符串

分析:

由于rope的底层实现,insert,erase,get都是logn的

就是翻转不行,不是自己手写的打不了标记啊!!

怎么办?

答:同时维护一正一反两个rope……反转即交换两个子串……Orz……

区间循环位移?简单,拆成多个子串连起来就好了……

区间a变b b变c c变d …… z变a? 呃……维护26个rope?

区间和?滚蛋,那是线段树的活

区间kth?sorry,与数值有关的操作rope一概不支持……

5555 维修数列只能自己写了……

ACCode:

#include <cstdio> #include <ext/rope> #include <iostream> #include <algorithm> using namespace std; using namespace __gnu_cxx; crope a,b,tmp; char s[10]; int now,n,len,size; char str[2000000],rstr[2000000]; int main(){     scanf("%d",&n);     while(n--)    {         scanf("%s",s);         switch(s[0])        {             case ‘M‘:            {                scanf("%d",&now);                break;            }             case ‘P‘:            {                now--;                break;            }             case ‘N‘:            {                now++;                break;            }             case ‘G‘:            {                putchar(a[now]);                putchar(‘\n‘);                break;            }             case ‘I‘:            {                 scanf("%d",&size);                 len=a.length();                 for(int i=0;i<size;i++)                {                     do                    {                        str[i]=getchar();                    }                     while(str[i]==‘\n‘);                     rstr[size-i-1]=str[i];                 }                 rstr[size]=str[size]=‘\0‘;                 a.insert(now,str);                 b.insert(len-now,rstr);                 break;             }                           case ‘D‘:            {                 scanf("%d",&size);                 len=a.length();                 a.erase(now,size);                 b.erase(len-now-size,size);                 break;             }             case ‘R‘:            {                 scanf("%d",&size);                 len=a.length();                 tmp=a.substr(now,size);                 a=a.substr(0,now) + b.substr(len-now-size,size) + a.substr(now+size,len-now-size);                 b=b.substr(0,len-now-size)+tmp+b.substr(len-now,now);                 break;             }         }     }     return 0; }

 

 

END