Leo_h

2333

后缀数组模板

刘汝佳的代码会RE!!!被坑了3小时!

以下模板自带注释

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
#define maxn 100005
#define rank RANK
int n;
char str[maxn];
int cnt[maxn];
int t1[maxn],t2[maxn];
int sa[maxn];
int rank[maxn];
int height[maxn];
void build()
{
	int *x=t1,*y=t2;
	int m=26;
	for(int i=0;i<n;i++) cnt[x[i]=(str[i]-'a')]++;
	for(int i=1;i<m;i++) cnt[i]+=cnt[i-1];
	for(int i=n-1;i>=0;i--) sa[--cnt[x[i]]]=i;
	for(int k=1;k<=n;k<<=1)
	{
		int p=0;
		for(int i=n-1;i>=n-k;i--) y[p++]=i;
		for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
		//y[i]:position of the suffix having the i-th "next k num",similar to sa
		memset(cnt,0,sizeof(cnt));
		for(int i=0;i<p;i++) cnt[x[y[i]]]++;
		for(int i=1;i<m;i++) cnt[i]+=cnt[i-1];
		for(int i=p-1;i>=0;i--) sa[--cnt[x[y[i]]]]=y[i];
		//give a higher rank to suffix with larger second keyword when having same first one
		swap(x,y);//this time,y stores the information
		p=0;
		for(int i=0;i<n;i++)
			x[sa[i]]= (!i||y[sa[i]]!=y[sa[i-1]]||y[sa[i]+k]!=y[sa[i-1]+k]) ? \
				p++:p-1;
		if(p>=n) break;
		m=p;
	}
	for(int i=0;i<n;i++) rank[sa[i]]=i;
	int k=0;
	for(int i=0;i<n;i++)
	{
		if(k) k--;
		if(rank[i]==0) continue;
		int j=sa[rank[i]-1];
		while(str[i+k]==str[j+k]) k++;
		height[rank[i]]=k;
	}
	
}
int main()
{
 	//freopen(".in","r",stdin);
 	scanf("%d",&n);
 	scanf("%s",str);
 	build();
 	for(int i=0;i<n;i++) printf("%d ",sa[i]);
 	putchar('\n');
 	for(int i=0;i<n;i++) printf("%d ",rank[i]+1);
 	putchar('\n');
 	for(int i=1;i<n;i++) printf("%d ",height[i]);
}


阅读更多
版权声明:本文为博主原创文章,转载请注明源网址blog.csdn.net/leo_h1104 https://blog.csdn.net/Leo_h1104/article/details/51481446
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

后缀数组模板

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭