【JZOJ 4876】 基因突变

Description

707获取了人类的基因信息并尝试对基因进行实验。他发现可以把人类的基因看做一个只包含小写字母的字符串，并定义从头开始任意长度的基因为“源头基因”人类身上与源头基因完全匹配的片段越多，这个人就越容易被控制。于是707就开始了他邪恶的计划……
作为人类卫士的射手ZMiG自然不会让707得逞，他决定拯救人类，现在他拿到了其中一个人被改造后的基因，他想请你统计一下它的基因中究竟有多少基因片段是可以与源头基因相匹配的
对于100%的数据，|S|<= 10^6

Analysis

考试的时候第一感觉扩展KMP，但是忘记怎么打了
普通KMP又想不到怎么做，于是本蒟蒻去打SA了！
结果倍增被卡常80分QAQ
SA很简单，答案是所有后缀跟原串的lcp之和，那么可以求出height数组，扫一遍
好吧上面的方法扯淡
正解普通KMP500B
然后设个f，可以发现f[i]=f[nxt[i]]+1
ans=sum(f[i])

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
using namespace std;
typedef long long ll;
const int N=1000010;
char s[N];
int nxt[N];
ll f[N];
int main()
{
    freopen("gene.in","r",stdin);
    freopen("gene.out","w",stdout);
    scanf("%s",s+1);
    int n=strlen(s+1),j=0;
    fo(i,2,n)
    {
        while(j && s[i]!=s[j+1]) j=nxt[j];
        nxt[i]=s[i]==s[j+1]?(++j):(j=0);
    }
    ll ans=0;
    f[1]=1;
    fo(i,1,n)
    {
        f[i]=f[nxt[i]]+1;
        ans+=f[i];
    }
    printf("%lld",ans);
    return 0;
}