【LGP5108】仰望半月的夜空

题目

我还会写(SA)(ST)表真是令人感动

发现这是一个思博题

我们开一个指针,标记一下当前合法的字典序最小的后缀排名在哪里,刚开始自然是(1)

我们发现这个后缀不能为我们提供(i)的长度我们就右移这个指针

之后我们二分+(St)表找到从这个后缀往右扩展的最大距离,查一下这里面最小的(sa)就好了

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const int maxn=300005;
inline int read() {
	char c=getchar();int x=0;while(c<'0'||c>'9') c=getchar();
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
char S[maxn];
int sa[maxn],het[maxn],rk[maxn],tp[maxn],tax[maxn];
int c[maxn],a[maxn];
int n,opt,m,sz;
int log_2[maxn],St[maxn][20],v[maxn][20];
inline int find(int x) {
	int l=1,r=sz;
	while(l<=r) {
		int mid=l+r>>1;
		if(c[mid]==x) return mid;
		if(c[mid]<x) l=mid+1;else r=mid-1;
	}
	return 0;
}
inline void qsort() {
	for(re int i=0;i<=m;i++) tax[i]=0;
	for(re int i=1;i<=n;i++) tax[rk[i]]++;
	for(re int i=1;i<=m;i++) tax[i]+=tax[i-1];
	for(re int i=n;i;i--) sa[tax[rk[tp[i]]]--]=tp[i];
}
inline int query(int l,int r) {
	int k=log_2[r-l+1];
	return min(St[l][k],St[r-(1<<k)+1][k]);
}
inline int ask(int l,int r) {
	int k=log_2[r-l+1];	
	return min(v[l][k],v[r-(1<<k)+1][k]);
}
int main() {
	opt=read(),n=read();
	if(opt==26) {scanf("%s",S+1);for(re int i=1;i<=n;i++) a[i]=S[i]-'a';}
	else for(re int i=1;i<=n;i++) a[i]=read();
	for(re int i=1;i<=n;i++) c[i]=a[i];
	std::sort(c+1,c+n+1);sz=std::unique(c+1,c+n+1)-c-1;
	for(re int i=1;i<=n;i++) a[i]=find(a[i]);
	for(re int i=1;i<=n;i++) rk[i]=a[i],tp[i]=i;
	m=sz;qsort();
	for(re int w=1,p=0;p<n;m=p,w<<=1) {
		p=0;
		for(re int i=1;i<=w;i++) tp[++p]=n-w+i;
		for(re int i=1;i<=n;i++) if(sa[i]>w) tp[++p]=sa[i]-w;
		qsort();
		for(re int i=1;i<=n;i++) std::swap(rk[i],tp[i]);
		rk[sa[1]]=p=1;
		for(re int i=2;i<=n;i++)
			rk[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&tp[sa[i]+w]==tp[sa[i-1]+w])?p:++p;
	}
	int k=0;
	for(re int i=1;i<=n;i++) {
		if(k) --k;
		int j=sa[rk[i]-1];
		while(a[i+k]==a[j+k]) ++k;
		het[rk[i]]=k;
	}
	for(re int i=1;i<=n;i++) St[i][0]=het[i];
	for(re int i=2;i<=n;i++) log_2[i]=log_2[i>>1]+1;
	for(re int j=1;j<=log_2[n];j++) 
		for(re int i=1;i+(1<<j)-1<=n;i++)
			St[i][j]=min(St[i][j-1],St[i+(1<<(j-1))][j-1]);
	for(re int i=1;i<=n;i++) v[i][0]=sa[i];
	for(re int j=1;j<=log_2[n];j++)
		for(re int i=1;i+(1<<j)-1<=n;i++)
			v[i][j]=min(v[i][j-1],v[i+(1<<(j-1))][j-1]);
	int now=1;
	for(re int i=1;i<=n;i++) {
		while(sa[now]+i-1>n) now++;
		int l=2,r=n-now+1;
		int ans=1;
		while(l<=r) {
			int mid=l+r>>1;
			if(query(now+1,now+mid-1)>=i) ans=mid,l=mid+1;
				else r=mid-1;
		}
		printf("%d ",ask(now,now+ans-1));
	}
	return 0;
}
原文地址:https://www.cnblogs.com/asuldb/p/10562757.html