首页 > 代码库 > 【BZOJ】1014: [JSOI2008]火星人prefix(splay+hash+二分+lcp)
【BZOJ】1014: [JSOI2008]火星人prefix(splay+hash+二分+lcp)
http://www.lydsy.com/JudgeOnline/problem.php?id=1014
被sb错调哭了QAQ。。。insert那里。。插入到第x个后边。。。我。。。。。。写成了第x个前面。。。。。。。。。。还调了!好!久!
QAQ
本题神lcp做法。。。。表示只会sa的height的离线。。。。。。。这种在线的我就QAQ做个忧伤的表情。。。
然后膜拜了题解。。。。好神。。splay维护区间。。。hash+二分维护lcp。。。。QAQ似乎是白书上说的么。。。
但是这种有概率的题这样搞真的好么。。。。。
因此取模那里直接抄人家的。。。因为不保证自己取的mod正确QAQ。。。。
然后就是在区间里乘上hash的权,因为乘法分配率,所以是可以区间维护的。
。。
#include <cstdio>#include <cstring>#include <cmath>#include <string>#include <iostream>#include <algorithm>#include <queue>#include <set>#include <map>using namespace std;typedef long long ll;#define rep(i, n) for(int i=0; i<(n); ++i)#define for1(i,a,n) for(int i=(a);i<=(n);++i)#define for2(i,a,n) for(int i=(a);i<(n);++i)#define for3(i,a,n) for(int i=(a);i>=(n);--i)#define for4(i,a,n) for(int i=(a);i>(n);--i)#define CC(i,a) memset(i,a,sizeof(i))#define read(a) a=getint()#define print(a) printf("%d", a)#define dbg(x) cout << (#x) << " = " << (x) << endl#define error(x) (!(x)?puts("error"):0)#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<‘0‘||c>‘9‘; c=getchar()) if(c==‘-‘) k=-1; for(; c>=‘0‘&&c<=‘9‘; c=getchar()) r=r*10+c-‘0‘; return k*r; }const int N=100005;const ll M=9875321;int P[N];struct node *null;struct node { node *f, *c[2]; int k, s, h; node(int _k=0) { k=_k; s=1; h=_k; f=c[0]=c[1]=null; } void setc(node *x, bool d) { c[d]=x; x->f=this; } bool d() { return f->c[1]==this; } void pushup() { s=c[0]->s+c[1]->s+1; h=(c[0]->h+((ll)k*P[c[0]->s])%M+((ll)c[1]->h*P[c[0]->s+1])%M)%M; }}*root;void rot(node *x) { node *f=x->f; bool d=x->d(); f->f->setc(x, f->d()); f->setc(x->c[!d], d); x->setc(f, !d); f->pushup(); if(f==root) root=x;}void splay(node *x, node *f=null) { while(x->f!=f) if(x->f->f==f) rot(x); else x->d()==x->f->d()?(rot(x->f), rot(x)):(rot(x), rot(x)); x->pushup();}node *sel(node *x, int k) { int s=x->c[0]->s; if(s==k) return x; if(s<k) return sel(x->c[1], k-s-1); return sel(x->c[0], k);}node *getrange(int l, int r) { splay(sel(root, l-1)); splay(sel(root, r+1), root); return root->c[1];}void ins(int k, int pos) { node *f=getrange(pos+1, pos); f->setc(new node(k), 0); splay(f->c[0]);}void fix(int k, int pos) { node *f=getrange(pos, pos); f->c[0]->k=k; f->c[0]->pushup(); splay(f->c[0]);}int gethash(int l, int r) { return getrange(l, r)->c[0]->h; }int ask(int x, int y) { int s=root->s-2; int l=1, r=min(s-x, s-y)+1, mid; while(l<=r) { mid=(l+r)>>1; if(gethash(x, x+mid-1)==gethash(y, y+mid-1)) l=mid+1; else r=mid-1; } return l-1;}void U(node *x) { if(x==null) return; U(x->c[0]); U(x->c[1]); x->pushup(); }void Pr(node *x) { if(x==null) return; Pr(x->c[0]); printf("%c", x->k+‘a‘); Pr(x->c[1]); }void D(node *x=root) { Pr(x); puts(""); U(x); Pr(x); }char s[N];int n;void build(int l, int r, node *&x) { if(l>r) return; int mid=(l+r)>>1; x=new node(s[mid]-‘a‘); if(l==r) return; build(l, mid-1, x->c[0]); build(mid+1, r, x->c[1]); if(l<=mid-1) x->c[0]->f=x; if(mid+1<=r) x->c[1]->f=x; x->pushup();}void init() { P[0]=1; for1(i, 1, 100000) P[i]=(P[i-1]*26)%M; null=new node; null->s=0; null->f=null->c[0]=null->c[1]=null; root=new node; root->setc(new node, 1); node *x; build(1, n, x); root->c[1]->setc(x, 0); root->c[1]->pushup(); root->pushup(); //D();}char rdchar() { char ret=getchar(); while(ret<‘a‘||ret>‘z‘) ret=getchar(); return ret;}int main() { scanf("%s", s+1); n=strlen(s+1); init(); int m=getint(); while(m--) { char c=getchar(); while(c!=‘Q‘&&c!=‘R‘&&c!=‘I‘) c=getchar(); if(c==‘Q‘) { int x=getint(), y=getint(); printf("%d\n", ask(x, y)); } else if(c==‘R‘) { int x=getint(); fix(rdchar()-‘a‘, x); } else if(c==‘I‘) { int x=getint(); ins(rdchar()-‘a‘, x); } } return 0;}
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、 询问。语法:Q x y,x, y均为正整数。功能:计算LCQ(x, y) 限制:1 <= x, y <= 当前字符串长度。 2、 修改。语法:R x d,x是正整数,d是字符。功能:将字符串中第x个数修改为字符d。限制:x不超过当前字符串长度。 3、 插入:语法:I x d,x是非负整数,d是字符。功能:在字符串第x个字符之后插入字符d,如果x = 0,则在字符串开头插入。限制:x不超过当前字符串长度。
Output
对于输入文件中每一个询问操作,你都应该输出对应的答案。一个答案一行。
Sample Input
7
Q 1 7
Q 4 8
Q 10 11
R 3 a
Q 1 7
I 10 a
Q 2 11
Sample Output
1
0
2
1
HINT
数据规模:
对于100%的数据,满足:
1、 所有字符串自始至终都只有小写字母构成。
2、 M <= 150,000
3、 字符串长度L自始至终都满足L <= 100,000
4、 询问操作的个数不超过10,000个。
对于第1,2个数据,字符串长度自始至终都不超过1,000
对于第3,4,5个数据,没有插入操作。
Source
【BZOJ】1014: [JSOI2008]火星人prefix(splay+hash+二分+lcp)