HihoCoder1445 后缀自动机二·重复旋律5

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
char s[maxn];
int len;
int T;
struct SAM{
    int last,cnt,nxt[maxn*2][26],fa[maxn*2],l[maxn*2];
    long long ans;
    void init()
    {
        last = cnt=1;
        memset(nxt[1],0,sizeof nxt[1]);
        fa[1]=0;ans=0;l[1]=0;
    }
    int inline newnode()
    {
        ++cnt;
        memset(nxt[cnt],0,sizeof nxt[cnt]);
        fa[cnt]=l[cnt]=0;
        return cnt;
    }
    void add(int c)
    {
        int p=last;
        int np=newnode();
        last=np;
        l[np]=l[p]+1;
        while(p&&!nxt[p][c])
        {
            nxt[p][c]=np;
            p = fa[p];
        }
        if(!p) fa[np]=1;
        else
        {
            int q = nxt[p][c];
            if(l[q]==l[p]+1) fa[np] = q;
            else
            {
                int nq = newnode();
                memcpy(nxt[nq],nxt[q],sizeof nxt[q]);
                fa[nq] = fa[q];
                l[nq] = l[p]+1;
                fa[np]=fa[q]=nq;
                while (nxt[p][c]==q)
                {
                    nxt[p][c]=nq;
                    p=fa[p];
                }
            }
        }
        ans+=(l[last]-l[fa[last]])*1ll;
    }
    void query()
    {
        int l=1,r=strlen(s+1);
        //scanf("%d%d",&l,&r);
        init();
        for(int i=l;i<=r;i++) add(s[i]-‘a‘);
        printf("%lld\n",ans);
    }
}sam;
int main()
{
    scanf("%s",s+1);
    sam.query();
    return 0;
}