版权声明:本文为博主原创文章,转载请附上原博客链接。 https://blog.csdn.net/Dale_zero/article/details/82899862
题目链接: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;
}