[BZOJ5417][SAM][线段树合并]NOI2018：你的名字

BZOJ5417

68pts：
我们考虑lim[i]为T中1-i的串的最长的在S中出现过的后缀，特别的，如果不存在这样的后缀，则lim[i]=0
lim[i]可以通过S的后缀自动机求出
具体的，我们在S上匹配T的每一个字符，如果无法匹配就在fail树上跳，直到能够匹配
我们设l[i]为T的后缀自动机中节点i的right集合能表示的最长的串，pos[i]表示该集合的字符串第一次出现的位置，tot为节点总个数，则每个节点的贡献可以表示为这个节点代表的集合的字符串个数减去能在S中匹配的个数
即$ans={\sum }_{i=2}^{tot}max(0,l[i]-max(l[fa[i]],lim[pos[i]]))$
就可以直接求了

100pts：
需要维护right集合，用线段树合并维护即可

Code：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
inline int read(){
	int res=0,f=1;char ch=getchar();
	while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
	while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
	return res*f;
}
const int N=3000008,M=2e7+5;
int n,root[N];
namespace segtree{
	int cnt=0,ls[M],rs[M];
	void insert(int &k,int l,int r,int pos){
		if(!k) k=++cnt;
		if(l==r) return;
		int mid=l+r>>1;
		if(pos<=mid) insert(ls[k],l,mid,pos);
		else insert(rs[k],mid+1,r,pos);
	}
	int merge(int x,int y){
		if(!x || !y) return x+y;
		int k=++cnt;
		ls[k]=merge(ls[x],ls[y]);
		rs[k]=merge(rs[x],rs[y]);
		return k;
	}
	bool query(int k,int l,int r,int ql,int qr){
		if(!k || ql>qr) return 0;
		if(ql<=l && r<=qr) return 1;
		int mid=l+r>>1;
		if(ql<=mid && query(ls[k],l,mid,ql,qr)) return 1;
		if(qr>mid && query(rs[k],mid+1,r,ql,qr)) return 1;
		return 0;
	}
}
using namespace segtree;
char s[N],t[N<<1];
namespace SAM1{
	int ch[N][26],l[N],fa[N],last,tot,pos[N];
	int c[N],a[N];
	inline void init(){last=tot=1;}
	inline void ins(int c){
		int p=last,np=++tot;last=np;l[np]=l[p]+1;pos[np]=1;
		for(;p && !ch[p][c];p=fa[p]) ch[p][c]=np;
		if(!p) fa[np]=1;
		else{
			int q=ch[p][c];
			if(l[q]==l[p]+1) fa[np]=q;
			else{
				int nq=++tot;l[nq]=l[p]+1;
				memcpy(ch[nq],ch[q],sizeof(ch[q]));
				fa[nq]=fa[q];fa[q]=fa[np]=nq;
				for(;p && ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
			}
		}
	}
	inline void work(){
		scanf("%s",s+1);
		init();int l1=strlen(s+1);n=l1;
		memset(c,0,sizeof(c));
		for(int i=1;i<=l1;i++) ins(s[i]-'a');
		for(int i=1;i<=tot;i++) ++c[l[i]];
		for(int i=1;i<=tot;i++) c[i]+=c[i-1];
		for(int i=1;i<=tot;i++) a[c[l[i]]--]=i;
		for(int i=tot,p;i;i--){
			p=a[i];
			if(pos[p]) insert(root[p],1,n,l[p]);
			root[fa[p]]=merge(root[fa[p]],root[p]);
		}
	}
}
int lim[N];
namespace SAM2{
	int rt;
	int ch[N][26],l[N],fa[N],last,tot,pos[N];
	inline void init(){tot=last=1,memset(ch[1],0,sizeof(int)*(26));}
    inline int New(){++tot;memset(ch[tot],0,sizeof(int)*(26));return tot;}
	inline void ins(int c){
		int p=last,np=New();last=np;pos[np]=l[np]=l[p]+1;
		for(;p && !ch[p][c];p=fa[p]) ch[p][c]=np;
		if(!p) fa[np]=1;
		else{
			int q=ch[p][c];
			if(l[q]==l[p]+1) fa[np]=q;
			else{
				int nq=New();l[nq]=l[p]+1;pos[nq]=pos[q];
				memcpy(ch[nq],ch[q],sizeof(ch[q]));
				fa[nq]=fa[q];fa[q]=fa[np]=nq;
				for(;p && ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
			}
		}
	}
	ll calc(int len){
		ll ret=0;
		for(int i=len+1;i<=tot;i++) ret+=max(0,l[i]-max(l[fa[i]],lim[pos[i]]));
		return ret;
	}
	inline void doit(int l,int r){
		init();int l2=strlen(t+1),p=1,kk=0;
		for(int i=1;i<=l2;i++){
			int c=t[i]-'a';
			ins(c);
			while(1){
				if(SAM1::ch[p][c] && query(root[SAM1::ch[p][c]],1,n,l+kk,r)) {++kk;p=SAM1::ch[p][c];break;}
				if(kk==0) break;
				--kk;
				if(kk==SAM1::l[SAM1::fa[p]]) p=SAM1::fa[p];
			}
			lim[i]=kk;
		}
	}
	inline void work(){
		int m=read();
		while(m--){
			scanf("%s",t+1);int l=read(),r=read();
			doit(l,r);
			cout<<calc(1)<<"\n";
		}
	}
}
int main(){
	SAM1::work();
	SAM2::work();
	return 0;
}