bzoj1014:[JSOI2008]火星人prefix

Description

  火星人最近研究了一种操作:求一个字串两个后缀的公共前缀。比方说,有这样一个字符串:madamimadam,
我们将这个字符串的各个字符予以标号:序号: 1 2 3 4 5 6 7 8 9 10 11 字符 m a d a m i m a d a m 现在,
火星人定义了一个函数LCQ(x, y),表示:该字符串中第x个字符开始的字串,与该字符串中第y个字符开始的字串
,两个字串的公共前缀的长度。比方说,LCQ(1, 7) = 5, LCQ(2, 10) = 1, LCQ(4, 7) = 0 在研究LCQ函数的过程
中,火星人发现了这样的一个关联:如果把该字符串的所有后缀排好序,就可以很快地求出LCQ函数的值;同样,
如果求出了LCQ函数的值,也可以很快地将该字符串的后缀排好序。 尽管火星人聪明地找到了求取LCQ函数的快速
算法,但不甘心认输的地球人又给火星人出了个难题:在求取LCQ函数的同时,还可以改变字符串本身。具体地说
,可以更改字符串中某一个字符的值,也可以在字符串中的某一个位置插入一个字符。地球人想考验一下,在如此
复杂的问题中,火星人是否还能够做到很快地求取LCQ函数的值。

Input

  第一行给出初始的字符串。第二行是一个非负整数M,表示操作的个数。接下来的M行,每行描述一个操作。操
作有3种,如下所示
1、询问。语法:Qxy,x,y均为正整数。功能:计算LCQ(x,y)限制:1<=x,y<=当前字符串长度。
2、修改。语法:Rxd,x是正整数,d是字符。功能:将字符串中第x个数修改为字符d。限制:x不超过当前字
符串长度。
3、插入:语法:Ixd,x是非负整数,d是字符。功能:在字符串第x个字符之后插入字符d,如果x=0,则在字
符串开头插入。限制:x不超过当前字符串长度

Output

  对于输入文件中每一个询问操作,你都应该输出对应的答案。一个答案一行。

Sample Input

madamimadam
7
Q 1 7
Q 4 8
Q 10 11
R 3 a
Q 1 7
I 10 a
Q 2 11

Sample Output

5
1
0
2
1

HINT

 

1、所有字符串自始至终都只有小写字母构成。

2、M<=150,000

3、字符串长度L自始至终都满足L<=100,000

4、询问操作的个数不超过10,000个。

对于第1,2个数据,字符串长度自始至终都不超过1,000

对于第3,4,5个数据,没有插入操作。

 

费尽千方百计终于把这道题A了……

题解:

插入和修改时用splay维护每个区间的hash(具体更新操作见代码)

LCQ可以二分求出:每次二分一个答案,判断时分别旋转看两个区间的hash值是否一样

 

注意:

1.update某区间hash值时不要忘了先赋成0(我是用指针写的)

2.不要忘了GetPow啊,检查时很不容易查出来,还以为自己是对的

3.由于前后都加了一个点,询问时若x=y得出的答案便会比标准答案多1,需要特判一下

4.询问操作x可能大于y,要判一下

5.bzoj不能用cin,bzoj不能用cin,bzoj不能用cin!(重要的事说三遍!!!为此我满篇的RE啊……说起来都是泪555)

 

代码:

  1 #include<cstdio>
  2 #include<iostream>
  3 #include<string.h>
  4 using namespace std;
  5 
  6 const int MAXN=140005;
  7 struct node
  8 {
  9     int d,size;
 10     int h;
 11     node *parent,*ch[2];       
 12 }pool[MAXN],*rf,*root;
 13 int cnt,pow[MAXN];
 14 void GetPow()
 15 {
 16     pow[0]=1;
 17     for(int i=1;i<MAXN;i++)
 18         pow[i]=(long long)pow[i-1]*137%10000007;
 19 }
 20 int size(node *p)
 21 {
 22     if(p) return p->size;
 23     return 0;
 24 }
 25 void update(node *p)
 26 {
 27     p->size=1+size(p->ch[0])+size(p->ch[1]);
 28     p->h=0;
 29     if(p->ch[0]) p->h=p->ch[0]->h;
 30     p->h=((long long)p->h*pow[1]+p->d)%10000007;
 31     if(p->ch[1]) p->h=((long long)p->h*pow[size(p->ch[1])]+p->ch[1]->h)%10000007;
 32 }
 33 void rotate(node *p,int type)
 34 {
 35     node *parent=p->parent,*son=p->ch[!type],*gp=p->parent->parent;
 36     parent->ch[type]=son;
 37     if(son) son->parent=parent;
 38     p->ch[!type]=parent;
 39     parent->parent=p;
 40     p->parent=gp;
 41     gp->ch[parent==gp->ch[1]]=p;
 42     update(parent);
 43     update(p);
 44     if(parent==root) root=p; 
 45 }
 46 void splay(node *p,node *target)
 47 {
 48     while(p->parent!=target)
 49     {
 50         if(p->parent->parent==target)
 51             rotate(p,p==p->parent->ch[1]);
 52         else
 53         {
 54             node *parent=p->parent,*gp=p->parent->parent;
 55             int f=parent==gp->ch[0];
 56             if(p==parent->ch[f]) rotate(p,f),rotate(p,!f);
 57             else rotate(parent,!f),rotate(p,!f);   
 58         }
 59     }
 60 }
 61 node *find(node *p,int k)
 62 {
 63     if(size(p->ch[0])>=k) return find(p->ch[0],k);
 64     else if(size(p->ch[0])==k-1) return p;
 65     else return find(p->ch[1],k-1-size(p->ch[0]));     
 66 }
 67 void insert(int x,node *newnode)
 68 {
 69     node *p=find(root,x);
 70     splay(p,rf);
 71     p=find(root,x+1);
 72     splay(p,root);
 73     p->ch[0]=newnode;
 74     newnode->parent=p;
 75     update(p);
 76     update(p->parent);
 77 }
 78 void change(int x,int d)
 79 {
 80     node *p=find(root,x);
 81     splay(p,rf);
 82     p->d=d;
 83     update(p);
 84 }
 85 bool query(int x,int y,int dis)
 86 {
 87     if(dis==-1) return true;
 88     node *p=find(root,x-1);
 89     splay(p,rf);
 90     p=find(root,x+dis+1);
 91     splay(p,root);
 92     int hash1=p->ch[0]->h;
 93     p=find(root,y-1);
 94     splay(p,rf);
 95     p=find(root,y+dis+1);
 96     splay(p,root);
 97     if(p->ch[0]->h==hash1) return true;
 98     return false;
 99 }
100 int Query(int x,int y)
101 {
102     if(x>y) swap(x,y);
103     int s=-1,t=size(root)-y-1,mid;
104     while(s<t)
105     {
106         mid=(s+t+1)/2;
107         if(query(x,y,mid)) s=mid;
108         else t=mid-1;
109     }
110     return s+1;
111 }
112 
113 int main()
114 {
115     char ch1[100000],sh[1],sh1[1];
116     int i,m,x,y,len;
117     node *p;
118     scanf("%s",ch1);
119     len=strlen(ch1);
120     
121     GetPow();
122     rf=&pool[++cnt];
123     root=&pool[++cnt];
124     root->d=root->h=0;
125     root->size=1;
126     root->parent=rf;
127     rf->ch[0]=root;
128     for(i=0;i<len;i++)
129     {
130         p=&pool[++cnt];
131         p->d=p->h=ch1[i]-'a'+1;
132         p->size=1;
133         root->ch[1]=p;
134         p->parent=root;
135         splay(p,rf);
136     }
137     p=&pool[++cnt];
138     p->d=p->h=0;p->size=1;
139     root->ch[1]=p;
140     p->parent=root;
141     splay(p,rf);
142     
143     scanf("%d",&m);
144     while(m --> 0)
145     {
146         scanf("%s",sh);
147         if(sh[0]=='Q')
148         {
149             scanf("%d%d",&x,&y);
150             x++;y++;
151             if(x!=y) printf("%d\n",Query(x,y));
152             else printf("%d\n",size(root)-x);
153         }
154         else if(sh[0]=='R')
155         {    
156             scanf("%d",&x);scanf("%s",sh1);
157             x++;
158             change(x,sh1[0]-'a'+1);
159         }
160         else
161         {
162             scanf("%d",&x);scanf("%s",sh1);
163             x++;
164             p=&pool[++cnt];
165             p->d=p->h=sh1[0]-'a'+1;
166             p->size=1;
167             insert(x,p);
168         }
169     }
170     
171     return 0;
172 }
View Code

 

既然选择了远方,便只顾风雨兼程
posted @ 2017-06-07 12:54  秋千旁的蜂蝶~  阅读(228)  评论(0编辑  收藏  举报