UOJ #35. 后缀排序 后缀自动机构造后缀树维护后缀数组

把串倒过来建后缀自动机得到的就是parent树就是后缀树

后缀树的遍历一发就得到后缀数组了

转移版 便于理解

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100;

int n;
char str[N];

namespace SAM
{
	int trans[N<<1][26],par[N<<1],mx[N<<1];
	int sz(1),root(1),suff(1);
	bool book[N<<1];
	int pos[N<<1];
	
	void insert(int x,int now)
	{
		int p=suff,np=++sz;
		mx[np]=mx[p]+1;
		pos[np]=now,book[np]=1;
		while(p && !trans[p][x])
			trans[p][x]=np,p=par[p];
		if(!p) par[np]=root;
		else
		{
			int q=trans[p][x];
			if(mx[q]==mx[p]+1) par[np]=q;
			else
			{
				int nq=++sz;
				mx[nq]=mx[p]+1,pos[nq]=pos[q];
				memcpy(trans[nq],trans[q],sizeof(trans[q]));
				par[nq]=par[q];
				par[q]=par[np]=nq;
				while(p && trans[p][x]==q)
					trans[p][x]=nq,p=par[p];
			}
		}
		suff=np;
	}
	
	void build()
	{
		register int i;
		for(i=n;i;--i)
			insert(str[i]-'a',i);
	}
}

namespace SA
{
	int trans[N<<1][26];
	
	int cnt,sa[N],rk[N],height[N];
	
	void dfs(int u)
	{
		if(!u) return ;
		if(SAM::book[u])
			sa[rk[SAM::pos[u]]=++cnt]=SAM::pos[u];
		for(int i=0;i<26;++i)
			dfs(trans[u][i]);
	}
	
	void get_sa()
	{
		register int i;
		for(i=2;i<=SAM::sz;++i)
			trans[SAM::par[i]][str[ SAM::pos[i]+SAM::mx[SAM::par[i]] ]-'a']=i;
		dfs(SAM::root);
	}
	
	void get_height()
	{
		register int i,j,k(0);
		for(i=1;i<=n;++i)
		{
			j=rk[i]-1;
			while(str[i+k]==str[sa[j]+k]) k++;
			height[rk[i]]=k;
			if(k) k--;
		}
	}
}

int main()
{
	scanf("%s",str+1);
	n=strlen(str+1);
	SAM::build();
	SA::get_sa();
	SA::get_height();
	register int i;
	for(i=1;i<=n;++i)
		printf("%d ",SA::sa[i]);puts("");
	for(i=2;i<=n;++i)
		printf("%d ",SA::height[i]);puts("");
	return 0;
}

邻接表版 稍快一点

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100;

int n;
char str[N];

namespace SAM
{
	int trans[N<<1][26],par[N<<1],mx[N<<1];
	int sz(1),root(1),suff(1);
	bool book[N<<1];
	int pos[N<<1];
	
	void insert(int x,int now)
	{
		int p=suff,np=++sz;
		mx[np]=mx[p]+1;
		pos[np]=now,book[np]=1;
		while(p && !trans[p][x])
			trans[p][x]=np,p=par[p];
		if(!p) par[np]=root;
		else
		{
			int q=trans[p][x];
			if(mx[q]==mx[p]+1) par[np]=q;
			else
			{
				int nq=++sz;
				mx[nq]=mx[p]+1,pos[nq]=pos[q];
				memcpy(trans[nq],trans[q],sizeof(trans[q]));
				par[nq]=par[q];
				par[q]=par[np]=nq;
				while(p && trans[p][x]==q)
					trans[p][x]=nq,p=par[p];
			}
		}
		suff=np;
	}
	
	void build()
	{
		register int i;
		for(i=n;i;--i)
			insert(str[i]-'a',i);
	}
}

namespace SA
{
	int last[N<<1],ecnt;
	struct EDGE{int to,nt;}e[N<<1];
	inline void add(int u,int v)
	{e[++ecnt]=(EDGE){v,last[u]};last[u]=ecnt;}
	
	int cnt,sa[N],rk[N],height[N];
	
	void dfs(int u)
	{
		if(SAM::book[u])
			sa[rk[SAM::pos[u]]=++cnt]=SAM::pos[u];
		for(int i=last[u];i;i=e[i].nt)
			dfs(e[i].to);
	}
	
	int buc[30],link[N<<1],st[N<<1];
	
	void get_sa()
	{
		register int i;
		for(i=2;i<=SAM::sz;++i)
			link[i]=str[ SAM::pos[i]+SAM::mx[SAM::par[i]] ]-'a',
			buc[link[i]]++;
		for(i=1;i<26;++i) buc[i]+=buc[i-1];
		for(i=SAM::sz;i>1;--i) st[buc[link[i]]--]=i;
		for(i=SAM::sz-1;i;--i)
			add(SAM::par[st[i]],st[i]);
		dfs(SAM::root);
	}
	
	void get_height()
	{
		register int i,j,k(0);
		for(i=1;i<=n;++i)
		{
			j=rk[i]-1;
			while(str[i+k]==str[sa[j]+k]) k++;
			height[rk[i]]=k;
			if(k) k--;
		}
	}
}

int main()
{
	scanf("%s",str+1);
	n=strlen(str+1);
	SAM::build();
	SA::get_sa();
	SA::get_height();
	register int i;
	for(i=1;i<=n;++i)
		printf("%d ",SA::sa[i]);puts("");
	for(i=2;i<=n;++i)
		printf("%d ",SA::height[i]);puts("");
	return 0;
}