SPOJ705 SPOJ694 Distinct Substrings SAM

SPOJ705
SPOJ694

题意：
给你一个字符串，问你有多少个本质不同的字符串。其中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;
}