【BZOJ】1014: [JSOI2008]火星人prefix（splay+hash+二分+lcp）

http://www.lydsy.com/JudgeOnline/problem.php?id=1014

题意：支持插入一个字符、修改一个字符，查询lcp。（总长度<=100000, 操作<=150000）

#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;

}　

被sb错调哭了QAQ。。。insert那里。。插入到第x个后边。。。我。。。。。。写成了第x个前面。。。。。。。。。。还调了！好！久！

QAQ

本题神lcp做法。。。。表示只会sa的height的离线。。。。。。。这种在线的我就QAQ做个忧伤的表情。。。

然后膜拜了题解。。。。好神。。splay维护区间。。。hash+二分维护lcp。。。。QAQ似乎是白书上说的么。。。

但是这种有概率的题这样搞真的好么。。。。。

因此取模那里直接抄人家的。。。因为不保证自己取的mod正确QAQ。。。。

然后就是在区间里乘上hash的权，因为乘法分配率，所以是可以区间维护的。

。。