Jzoj4384 Hashit

最新推荐文章于 2022-09-16 23:24:52 发布

原创最新推荐文章于 2022-09-16 23:24:52 发布 · 596 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#后缀平衡树 #后缀自动机 #字符串 #扩展的灰

OI 同时被 3 个专栏收录

321 篇文章

订阅专栏

Jzoj

227 篇文章

订阅专栏

数据结构

134 篇文章

订阅专栏

本文介绍了一种使用后缀平衡树解决字符串问题的方法。通过实战练习，实现了字符串操作及子串计数的功能。利用哈希技术和set数据结构，有效地处理字符串的插入与删除操作，并实时更新两两不同的子串数量。

你有一个字符串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);
	}
}