洛谷P4070 生成魔咒

题意：给定字符串，求每个前缀的本质不同的子串数量。字符集1e9。

解：在线构造后缀自动机并统计答案。

答案就是∑len[i] - len[fail[i]]

每次增加的时候，至多对三个节点有影响。然而把Q分裂为nQ本质不同的子串数没变。

于是增加的只有len[np] - len[fail[np]]

map维护转移边。

 #include <cstdio>
 #include <map>

 typedef long long LL;
 const int N = ;

 int len[N], fail[N], last, top;
 LL ans;
 std::map<int, int> mp[N];

 inline void init() {
     top = last = ;
     return;
 }

 inline void insert(int f) {
     int p = last, np = ++top;
     last = np;
     len[np] = len[p] + ;
     while(p && !mp[p].count(f)) {
         mp[p][f] = np;
         p = fail[p];
     }
     //printf("p = %d \n", p);
     if(!p) {
         fail[np] = ;
         ans += len[np];
         //printf("1 : ans += %d \n", len[np]);
     }
     else {
         int Q = mp[p][f];
         //printf("Q = %d \n", Q);
         if(len[Q] == len[p] + ) {
             fail[np] = Q;
             ans += len[np] - len[Q];
             //printf("2 : ans += %d \n", len[np] - len[Q]);
         }
         else {
             int nQ = ++top;
             len[nQ] = len[p] + ;
             fail[nQ] = fail[Q];
             fail[Q] = fail[np] = nQ;
             mp[nQ] = mp[Q];
             while(mp[p].count(f) && mp[p][f] == Q) {
                 mp[p][f] = nQ;
                 p = fail[p];
             }
             ans += len[np] - len[nQ];
         }
     }
     return;
 }

 int main() {
     int n;
     scanf("%d", &n);
     init();
     for(int i = , x; i <= n; i++) {
         scanf("%d", &x);
         insert(x);
         printf("%lld\n", ans);
     }

     return ;
 }

AC代码