后缀数组SA-IS模板

最新推荐文章于 2025-05-27 13:41:40 发布

原创最新推荐文章于 2025-05-27 13:41:40 发布 · 1.1k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#后缀数组 #字符串 #SAIS #扩展的灰

OI 同时被 3 个专栏收录

321 篇文章

订阅专栏

字符串

42 篇文章

订阅专栏

----后缀数组

11 篇文章

订阅专栏

本文介绍了后缀数组SA-IS算法，强调其在解决字符串问题时的高效性，并推荐了若干篇相关优质博客和论文资源，帮助读者深入理解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

.
SAIS真的好快啊，在UOJ能卡进第二页
推荐几个讲的比较好的博客
链接1
链接2
链接3
和一篇讲了各种后缀数组算法的论文
Code:

#pragma GCC optimize("O3")
#pragma G++ optimize("O3")
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 100010
using namespace std;
const int BufferSize = 1180000;
char buffer[BufferSize];
char *out_tail = buffer;
inline void putint(int x)
{
	if (!x) *out_tail++ = '0';
	else
	{
		register char s_pool[6], *s_tail = s_pool;
		while (x != 0) *s_tail++ = x % 10 + '0', x /= 10;
		while (s_tail-- != s_pool) *out_tail++ = *s_tail;
	}
	*out_tail++ = ' ';
}
char S[N]; int n,m;
int s[N<<1],t[N<<1],H[N],sa[N],r[N],p[N],c[N],w[N];
inline int trans(int n,const char* S){
	int m=*max_element(S+1,S+1+n);
	for(int i=1;i<=n;++i) r[S[i]]=1;
	for(int i=1;i<=m;++i) r[i]+=r[i-1];
	for(int i=1;i<=n;++i) s[i]=r[S[i]];
	return r[m];
}
#define ps(x) sa[w[s[x]]--]=x
#define pl(x) sa[w[s[x]]++]=x
inline void radix(int* v,int* s,int* t,int n,int m,int n1){
	memset(sa,0,n+1<<2); memset(c,0,m+1<<2);
	for(int i=1;i<=n;++i) ++c[s[i]];
	for(int i=1;i<=m;++i) w[i]=c[i]+=c[i-1];
	for(int i=n1;i;--i) ps(v[i]);
	for(int i=1;i<=m;++i) w[i]=c[i-1]+1;
	for(int i=1;i<=n;++i) if(sa[i]>1 && t[sa[i]-1]) pl(sa[i]-1);
	for(int i=1;i<=m;++i) w[i]=c[i];
	for(int i=n;i;--i) if(sa[i]>1 && !t[sa[i]-1]) ps(sa[i]-1);
}
inline void SAIS(int n,int m,int* s,int* t,int* p){
	int n1=0,ch=r[1]=0,*s1=s+n; t[n]=0;
	for(int i=n-1;i;--i) t[i]=s[i]==s[i+1]?t[i+1]:s[i]>s[i+1];
	for(int i=2;i<=n;++i) r[i]=t[i-1]&&!t[i]?(p[++n1]=i,n1):0;
	radix(p,s,t,n,m,n1);
	for(int i=1,x,y;i<=n;++i) if(x=r[sa[i]]){
		if(ch<=1 || p[x+1]-p[x]!=p[y+1]-p[y]) ++ch;
		else for(int j=p[x],k=p[y];j<=p[x+1];++j,++k)
			if((s[j]<<1|t[j])^(s[k]<<1|t[k])){ ++ch; break; }
		s1[y=x]=ch;
	}
	if(ch<n1) SAIS(n1,ch,s1,t+n,p+n1);
	else for(int i=1;i<=n1;++i) sa[s1[i]]=i;
	for(int i=1;i<=n1;++i) s1[i]=p[sa[i]];
	radix(s1,s,t,n,m,n1);
}
inline void SA(int n,const char* S){
	int m=trans(++n,S);	SAIS(n,m,s,t,p);
	for(int i=1;i<n;++i) r[sa[i]=sa[i+1]]=i;
	for(int i=1,j,k=0;i<n;++i) if(r[i]>1){
		for(j=sa[r[i]-1];S[i+k]==S[j+k];++k);
		if(H[r[i]]=k) --k;
	}
}
int main(){
	gets(S+1); n=strlen(S+1); SA(n,S);
	for (int i = 1; i <= n; ++i) putint(sa[i]);
	*out_tail++ = '\n';
	for (int i = 2; i <= n; ++i)  putint(H[i]);
	fwrite(buffer, 1, out_tail - buffer, stdout);
}