版权声明:本文为博主原创文章,可以转载但是必须声明版权。 https://blog.csdn.net/forever_shi/article/details/84788624
题意:
给你一个字符串,问你有多少个本质不同的字符串。其中SPOJ705是加强版,20组询问,每组n<=50000。
题解:
这个题异常坑爹!!!我调这个题调了一上午,心态爆炸。这个加强版没有说字符集大小,我以为是26,结果发现不是!
回归正题,这个题的做法其实就是建出SAM,然后对于每个点,用它的串长减去parent树上它父节点的串长就行了,这样就可以不重不漏的计数了。
代码:
#include <bits/stdc++.h>
using namespace std;
int T,n,fa[2000010],ch[100010][100],len[2000010],cnt=1,rt=1,lst=1;
long long ans;
char s[1000010];
inline void insert(int x)
{
int cur=++cnt,pre=lst;
lst=cur;
len[cur]=len[pre]+1;
for(;pre&&!ch[pre][x];pre=fa[pre])
ch[pre][x]=cur;
if(!pre)
fa[cur]=rt;
else
{
int ji=ch[pre][x];
if(len[ji]==len[pre]+1)
fa[cur]=ji;
else
{
int gg=++cnt;
len[gg]=len[pre]+1;
memcpy(ch[gg],ch[ji],sizeof(ch[ji]));
fa[gg]=fa[ji];
fa[ji]=fa[cur]=gg;
for(;pre&&ch[pre][x]==ji;pre=fa[pre])
ch[pre][x]=gg;
}
}
}
int main()
{
scanf("%d",&T);
while(T--)
{
cnt=1,rt=1,lst=1;
ans=0;
scanf("%s",s+1);
n=strlen(s+1);
for(int i=1;i<=n;++i)
insert(s[i]-'A');
for(int i=1;i<=cnt;++i)
ans+=len[i]-len[fa[i]];
for(int i=0;i<=cnt;++i)
{
fa[i]=len[i]=0;
memset(ch[i],0,sizeof(ch[i]));
}
printf("%lld\n",ans);
}
return 0;
}