桂林电子科技大学第三届ACM程序设计竞赛 A 题串串 后缀数组

版权声明：欢迎转载，转载请注明出处，如有错误，还望指出，谢谢 博客地址：https://blog.csdn.net/lanyanzhiji123asd https://blog.csdn.net/lanyanzhiji123asd/article/details/89496004

 比赛链接   https://ac.nowcoder.com/acm/contest/558#question

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<set>
#include<queue>
using namespace std;
#define ll long long
int n,m,kk;
const int ding=1e5+5;
const int INF=0x3f3f3f3f;
int sa[ding],x[ding],y[ding],c[ding],r[ding],rankk[ding],height[ding];

void get_SA()
{
	int i,k;
//	memset(x,0,sizeof(x));//因为我是以a来作为终止符，所以后面的空 应该设置成-1 
//	memset(y,0,sizeof(y));
	memset(c,0,sizeof(c));//清空桶；
	for(i=0;i<n;i++)	c[x[i]=r[i]]++; 
	for(i=1;i<m;i++)	c[i]+=c[i-1];
	for(i=n-1;i>=0;i--)	
	{
		sa[--c[x[i]]]=i;
	}
	int p=1;
	for(k=1;k<n;k<<=1,m=p)
	{
		p=0;
		for(i=n-k;i<n;i++)	y[p++]=i;//
		for(i=0;i<n;i++)	if(sa[i]>=k) y[p++]=sa[i]-k;//大于等于k 
		//sa是第一关键字的排名， 然后再根据第二关键字的排名来确定排名， 
		memset(c,0,sizeof(c));
		for(i=0;i<n;i++)	c[x[y[i]]]++;//统计当前字符串 
		for(i=1;i<m;i++)	c[i]+=c[i-1];//统计个数 
		for(i=n-1;i>=0;i--)		sa[--c[x[y[i]]]]=y[i];     //重置sa数组
		memcpy(y,x,sizeof(x));
		memset(x,0,sizeof(x));
//		swap(x,y);
		p=1;//从1开始编号 
		x[sa[0]]=0;
		//因为上面已经重置了x的值  所以下面是从1开始循环，然后是和前一个比较sa[i]和sa[i-1] 比较 
		for(i=1;i<n;i++) 
			x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?p-1:p++;
		
		//x数组里面记住的是x[i]=j  下标为i的后缀，排名为j
		//y数组的内容是y[i]=j 下标为j 的后缀排名为i 
		if(p>=n)break;
	}
//	for(i=1;i<n;i++)
//		cout<<sa[i]+1<<" ";
//	cout<<endl;
}

void get_height()
{
	int i,k=0,j;
	for(i=0;i<n;i++) rankk[sa[i]]=i;
	for(i=0;i<n-1;i++)
	{//为什么循环到n-1，因为我们在原始字符串后面加了a，所以这个后缀一定是最小的。 
//		cout<<"i   "<<i<<endl;
		if(k)k--;
		j=sa[rankk[i]-1];
		for(;r[i+k]==r[j+k];k++);
		height[rankk[i]]=k; 
	}
//	for(i=0;i<n-1;i++)
//		cout<<height[i]<<" ";
//	cout<<endl;
		
} 



int main()
{
 	std::ios::sync_with_stdio(false);
 	string str;
 	ll ans=0;
	int i;
	cin>>str; 
	n=str.length();
	for(i=0;i<n;i++)
		r[i]=(int)(str[i]-'a'+1);
	r[n]=0;
	n++;
	m=30;
	get_SA();
	get_height();
	for(i=2;i<n;i++)
	{
		ans+=n-sa[i]-height[i]-1;
//		cout<<ans<<endl;
	}
	ans+=n-sa[1]-1;
//	cout<<sa[1]<<endl;
	cout<<ans<<endl;

 return 0;
}