2018焦作网络赛 H String and Times

题目链接：https://nanti.jisuanke.com/t/31717

计算原串中，有多少子串的出现次数位于l和r之间，后缀自动机的题目，建立后缀自动机之后进行拓扑排序，将子节点出现的次数不断向它在parent树上的父节点追加更新，计算每个节点right集合的大小，得到的就是其代表字符串出现的次数，有时间写一篇SAM的学习笔记吧，希望能帮到像我一样刚入门的新手。。

还有这道题比较坑的一点就是，strlen函数不要在for循环中使用，strlen好像是O（n）计算串长的，串长特别大的话可能会超时。这道题就是因为这个T了几次。。还是菜啊

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
#define mod 1000000007
#define For(i,m,n) for(int i=m;i<=n;i++)
#define Dor(i,m,n) for(int i=m;i>=n;i--)
#define LL long long
#define lan(a,b) memset(a,b,sizeof(a))
#define sqr(a) a*a
using namespace std;



const int MAXN = 4e5+10;
char s[200005];
const int LetterSize = 26;

int tot, last,ch[MAXN][LetterSize],fa[MAXN],len[MAXN];
int sum[MAXN],tp[MAXN],cnt[MAXN];

void init()
{
    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;
    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;
}

int main()
{
    while(~scanf("%s",s))
    {
        init();
        int l,r;
        scanf("%d%d",&l,&r);
        int n=strlen(s);
        for(int i=0;i<n;i++)
            add(s[i]-'A');
        toposort();
        LL ans=0;
        for(int i=tot;i>=1;i--)
        {
            int p=tp[i],fp=fa[p];
            cnt[fp]+=cnt[p];
            if(cnt[p] <= r && cnt[p] >= l) ans+=len[p]-len[fp];
        }
        printf("%lld\n",ans);
    }
    return 0;
}