【解题报告】SPOJ - SUBST1 New Distinct Substrings 后缀数组

求字符串不同子串的个数：
不能字符串hash了
后缀数组解法：考虑排名为i的后缀，其对子串的贡献为n-sa[i]+1，但是我们会发现这里面有和之前重复的，所以要减去这些重复的，而height[i]就是sa[i]和sa[i-1]之间重复的个数。所以最后的答案就是对(n-sa[i]+1-height[i])求和

AC代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=5e4+10;
char s[maxn];
int y[maxn],x[maxn],cnt[maxn],sa[maxn],rk[maxn],height[maxn];
int n,m;
int k;
void get_SA()
{
	memset(cnt,0,sizeof(cnt));
	m=300;
	for (int i=1;i<=n;++i) ++cnt[x[i]=s[i]];
	//x[i]是第i个元素的第一关键字 
    for (int i=2;i<=m;++i) cnt[i]+=cnt[i-1]; 
    for (int i=n;i>=1;--i) sa[cnt[x[i]]--]=i; 
    for (int k=1;k<=n;k<<=1)
    {
        int num=0;
        for (int i=n-k+1;i<=n;++i) y[++num]=i;
        //y[i]表示第二关键字排名为i的数，第一关键字的位置 
        for (int i=1;i<=n;++i) if (sa[i]>k) y[++num]=sa[i]-k;
        for (int i=1;i<=m;++i) cnt[i]=0;
        for (int i=1;i<=n;++i) ++cnt[x[i]];
        for (int i=2;i<=m;++i) cnt[i]+=cnt[i-1];//第一关键字排名为1~i的数有多少个 
        for (int i=n;i>=1;--i) sa[cnt[x[y[i]]]--]=y[i],y[i]=0;
        swap(x,y);
        x[sa[1]]=1;num=1;
        for (int i=2;i<=n;++i)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]) ? num : ++num;
        if (num==n) break;
        m=num;
    }
}
void get_height()
{
    int k=0;
    for (int i=1;i<=n;++i) rk[sa[i]]=i;
    for (int i=1;i<=n;++i)  
    {
        if (rk[i]==1) continue;//第一名height为0 
        if (k) --k;//h[i]>=h[i-1]+1;
        int j=sa[rk[i]-1];
        while (j+k<=n && i+k<=n && s[i+k]==s[j+k]) ++k;
        height[rk[i]]=k;//h[i]=height[rk[i]];
    }
}
void solve()
{
	long long ans=0;
	for(int i=1;i<=n;i++)
		ans+=(n-sa[i]+1-height[i]);
	printf("%lld\n",ans);
}
int main()
{
	int t;scanf("%d",&t);
	while(t--){
		scanf("%s",s+1);
		n=strlen(s+1);
		get_SA();
		get_height();
		solve();
	} 
}