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

时间限制:10000ms

单点时限:2000ms

内存限制:512MB

描述

小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一个音乐旋律被表示为一段数构成的数列。

现在小Hi想知道一部作品中出现了多少不同的旋律？

解题方法提示

输入

共一行，包含一个由小写字母构成的字符串。字符串长度不超过 1000000。

输出

一行一个整数，表示答案。

样例输入

aab

样例输出

5

---

---

---

SA可解，但是这次是学习SAM的解法
SAm建立的图中不会有一条路径是相同的字符串，所以所有字串的个数就是 每个状态的 manlen - minlen
而minlen 又等于 maxlen (fa[x])

#include <bits/stdc++.h>
using namespace std;
int last =1,tot=1;
struct NODE 
{
    int ch[26];
    int len,fa;
    NODE () {memset(ch,0,sizeof ch); len=fa=0;}

}dian[3000000];

inline void add(int c)
{
   int p = last; int np = last = ++tot; 
  // cout<<c+'a'<<endl;
   dian[np].len = dian[p].len+1;
   for( ; p && !dian[p].ch[c]; p=dian[p].fa)dian[p].ch[c] = np;
 //  cout<<" p: "<<p<<endl;
   if(!p) dian[np].fa = 1;
   else 
   {
    //   cout<<"HERE"<<endl;
       int q=dian[p].ch[c];
       if(dian[q].len == dian[p].len+1)dian[np].fa=q;
       else
       {
           int nq=++tot;
           dian[nq]=dian[q]; dian[nq].len=dian[p].len+1;
           dian[q].fa=dian[np].fa=nq;
           for(;p&&dian[p].ch[c]==q;p=dian[p].fa)dian[p].ch[c]=nq;
       }
   }



}
int main()
{
    string s;cin>>s; int len=s.length();
    for (int i=0;i<len;i++)add(s[i]-'a');
    long long ans = 0;
    //cout<<tot<<endl;
    for (int i=2;i<=tot;i++)ans+=dian[i].len - dian[dian[i].fa].len;
    cout<<ans;
}