后缀自动机

模板

struct SAM{
    static const int MAXN = 100001<<1;//大小为字符串长度两倍
    static const int LetterSize = 26;

    int tot, last, ch[MAXN][LetterSize], fa[MAXN], len[MAXN];
    int sum[MAXN], tp[MAXN], cnt[MAXN]; //sum,tp用于拓扑排序，tp为排序后的数组
    //tp 长度拓扑序后的位置
    //cnt
    /*1.对于每个节点，父节点的接受串是该节点的接受串的后缀，而且是最长的那个。
    所以呢，用该节点的len减去父节点的fa.len，就是该节点表示的串中父节点不能表示的串个数。
    当某节点的计数到达k次时，就说明有len-fa.len个不同子串计数到达k次了。
    （同理，当插入这个节点，出现的新不同子串个数就是len-fa.len。这也是很有用的性质。）

    2.每个节点表示串出现的次数，是这个节点子树上有效节点的总数。
    （有效节点指插入该字符产生的那个节点，也就是代码中extend中的np节点）。
    这样我们初始化有效节点为1，然后拓扑排序dp，就可以O(n)得到每个节点出现次数。*/
    void init( void){
        last = tot = 1;
        len[1] = 0;
        memset(ch,0,sizeof ch);
        memset(fa,0,sizeof fa);
        memset(cnt,0,sizeof cnt);
    }

    void add( int x){
        int p = last, np = last = ++tot;
        len[np] = len[p] + 1, cnt[last] = 1;
        memset( ch[np], 0, sizeof ch[np]);
        while( p && !ch[p][x]) ch[p][x] = np, p = fa[p];
        if( p == 0)
            fa[np] = 1;
        else{
            int q = ch[p][x];
            if( len[q] == len[p] + 1)
                fa[np] = q;
            else{
                int nq = ++tot;
                memcpy( ch[nq], ch[q], sizeof ch[q]);
                len[nq] = len[p] + 1, fa[nq] = fa[q], fa[q] = fa[np] = nq;
                while( p && ch[p][x] == q)  ch[p][x] = nq, p = fa[p];
            }
        }
    }

    void toposort(){
        for(int i = 1; i <= len[last]; i++)   sum[i] = 0;
        for(int i = 1; i <= tot; i++)   sum[len[i]]++;
        for(int i = 1; i <= len[last]; i++)   sum[i] += sum[i-1];
        for(int i = 1; i <= tot; i++)   tp[sum[len[i]]--] = i;
        for(int i = tot; i; i--)   cnt[fa[tp[i]]] += cnt[tp[i]];
    }

} sam;