[SAM] hihoCoder1445 后缀自动机二·重复旋律5

题意

给出一个只含小写字母的串 s ,求 s 的不同的子串总数。

|s|≤1000000

题解

SAM 最水的模板题。

直接建出 SAM ,把所有状态的 max(A)−min(A)+1 全部加起来即是答案。应该不需要解释。

实现的话可以在 Extend 过程中维护，这样写比较有价值，最后扫一趟太丑啦…

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1000005;
struct node{
node *par,*ch[26]; int _max;
node(int t1=0){ _max=t1; par=0; memset(ch,0,sizeof(ch)); }
int cnt(){ return _max-par->_max; }
} *root, *last;
typedef node* P_node;
long long ans;
void Extend(char x){
P_node p=last, np=new node(p->_max+1);
while(p&&p->ch[x]==0) p->ch[x]=np, p=p->par;
if(!p) np->par=root; else{
P_node q=p->ch[x];
if(q->_max==p->_max+1) np->par=q; else{
P_node nq=new node(p->_max+1);
memcpy(nq->ch,q->ch,sizeof(q->ch));
ans-=q->cnt(); //*
nq->par=q->par; q->par=nq; np->par=nq;
ans+=q->cnt()+nq->cnt(); //*
while(p&&p->ch[x]==q) p->ch[x]=nq, p=p->par;
}
}
ans+=np->cnt(); //*
last=np;
}
char s[maxn];
int n;
int main(){
freopen("hiho1445.in","r",stdin);
freopen("hiho1445.out","w",stdout);
scanf("%s",s+1); n=strlen(s+1);
root=last=new node(0);
for(int i=1;i<=n;i++) Extend(s[i]-'a');
printf("%lld\n",ans);
return 0;
}