Jzoj4384 Hashit

你有一个字符串S,最开始为空,要求支持两种操作

在S后面加入字符c

删除S最后一个字符

每次操作询问S有多少个两两不同子串

应该本来应该用SAM+Trie离线做的,然而为了练一下后缀平衡树就写了

其实也很好写,用哈希比较一下就好了,可以用set实现,开一个数组存每个后缀对应的节点就好

求height也可以用哈希

#pragma GCC opitmize("O3")
#pragma G++ opitmize("O3")
#include<set>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define LL long long
using namespace std;
char s[100010]; int n=0;
LL h[100010],bas[100010]={1};
struct suffix{ int x; };
inline LL gH(int l,int r){ return h[r]-h[l-1]*bas[r-l+1]; }
inline int lcp(int x,int y){
	if(x==y) return 0; 
	int l=-1,r=min(x,y)-1;
	for(int m;l<r;){
		m=l+r+1>>1;
		if(gH(x-m,x)==gH(y-m,y)) l=m;
		else r=m-1;
	}
	return l+1;
}
inline bool operator< (suffix a,suffix b){
	int c=lcp(a.x,b.x); return s[a.x-c]<s[b.x-c];
}
multiset<suffix> w;
multiset<suffix>::iterator c[100010],p,q;
int main(){
	int ans=0;
	for(int i=1;i<=100000;++i) bas[i]=bas[i-1]*27;
	for(char o;;){
		o=getchar();
		if(o=='-'){
			p=q=c[n]; ++p; --q;
			ans-=n-lcp(p->x,n)-lcp(n,q->x)+(p->x==n || q->x==n?0:lcp(p->x,q->x));
			w.erase(c[n--]); 
		}
		else if(o>='a' && o<='z'){
			s[++n]=o;  h[n]=h[n-1]*27+o-'a';
			p=q=c[n]=w.insert((suffix){n}); ++p; --q; 
			ans+=n-lcp(p->x,n)-lcp(n,q->x)+(p->x==n || q->x==n?0:lcp(p->x,q->x));
		} else break;
		printf("%d\n",ans);
	}
}

posted @ 2017-12-10 13:16  扩展的灰(Extended_Ash)  阅读(197)  评论(0编辑  收藏  举报