复习后缀数组

Posted 2020-12-02 linda-fcj

贴模板，复习注意事项

补一句，SA是$O(N(logN))$的，暴力是$O(N(log^2(N)))$的，不卡的话，实在忘记了，就暴力，实惠又划算

CODE:

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1e5+10;
int sa[maxn],tmp[maxn],hei[maxn],x[maxn],y[maxn],c[maxn],f[maxn][21],bin[21];
char s[maxn];

inline void SA(int n,int m)
{
	int p=0;
	for(int i=1;i<=m;i++) c[i]=0;
	for(int i=1;i<=n;i++) c[x[i]=s[i]]++;
	for(int i=1;i<=m;i++) c[i]+=c[i-1];
	for(int i=n;i>=1;i--) sa[c[x[i]]--]=i;//倒着
	for(int k=1;k<=n && p<=n;k<<=1)
	{
		p=0;
		for(int i=n-k+1;i<=n;i++) y[++p]=i;//后面的不足k
		for(int i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;
		for(int i=1;i<=m;i++) c[i]=0;
		for(int i=1;i<=n;i++) c[x[y[i]]]++;
		for(int i=1;i<=m;i++) c[i]+=c[i-1];
		for(int i=n;i>=1;i--) sa[c[x[y[i]]]--]=y[i];//倒着
		
		for(int i=1;i<=n;i++) tmp[i]=x[i];
		for(int i=1;i<=n;i++) x[i]=y[i];
		for(int i=1;i<=n;i++) y[i]=tmp[i];
//		swap(x,y);
		x[sa[1]]=p=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])?p:++p;
		m=p;
	}
	
	int k=0;
	for(int i=1;i<=n;i++)
	{
		if(x[i]==1) continue;
		if(k) k--;//>=hei[i-1]-1
		int j=sa[x[i]-1];//上一个排名
		while(i+k<=n && j+k<=n && s[i+k]==s[j+k]) k++;
		hei[x[i]]=k;
	}
}

inline void pr(int l,int r)
{
	for(int i=l;i<=r;i++) printf("%c",s[i]);
	printf("
");
}

int main()
{
	scanf("%s",s+1);
	int n=strlen(s+1);
	SA(n,1000);
	return 0;
}