HDU3518 Boring counting【出现次数大于1且不重叠子串的个数 (后缀自动机)】

题目描述：

大意RT。

题目分析：

求出后缀自动机上每个节点endpos集合的最左和最右记为$L,R$。
如果$R-L\ge len[i]-len[fa[i]]$，那么这个点代表的所有子串都可行。
否则就只是$\max (R-L-len[fa[i]],0)$的那部分可行。

Code：

#include<bits/stdc++.h>
#define maxn 2005
#define maxc 26
using namespace std;
int T,n,ch[maxn][maxc],fa[maxn]={-1},len[maxn],last,sz,b[maxn],sa[maxn],L[maxn],R[maxn];
char s[maxn];
void extend(int c,int pos){
	int cur=++sz,p=last,q; len[last=cur]=len[p]+1;
	L[cur]=R[cur]=pos;
	for(;p!=-1&&!ch[p][c];p=fa[p]) ch[p][c]=cur;
	if(p==-1) fa[cur]=0;
	else if(len[q=ch[p][c]]==len[p]+1) fa[cur]=q;
	else{
		int clone=++sz; len[clone]=len[p]+1,fa[clone]=fa[q];
		L[clone]=R[clone]=pos;
		fa[cur]=fa[q]=clone,memcpy(ch[clone],ch[q],sizeof ch[q]);
		for(;p!=-1&&ch[p][c]==q;p=fa[p]) ch[p][c]=clone;
	}
}
void Clear(){
	memset(ch,0,maxc*(sz+1)*4),memset(b,0,(sz+1)<<2),sz=last=0;
}
int main()
{
	while(scanf("%s",s+1),s[1]!='#'){
		Clear(),n=strlen(s+1);
		for(int i=1;i<=n;i++) extend(s[i]-'a',i);
		for(int i=1;i<=sz;i++) b[len[i]]++;
		for(int i=1;i<=n;i++) b[i]+=b[i-1];
		for(int i=1;i<=sz;i++) sa[b[len[i]]--]=i;
		for(int i=sz,x;i>=1;i--)
			x=sa[i],L[fa[x]]=min(L[fa[x]],L[x]),R[fa[x]]=max(R[fa[x]],R[x]);
		int ans=0;
		for(int i=1;i<=sz;i++)
			if(R[i]-L[i]>=len[i]) ans+=len[i]-len[fa[i]];
			else ans+=max(R[i]-L[i]-len[fa[i]],0);
		printf("%d\n",ans);
	}
}