1. 程式人生 > >●BZOJ 4516 [Sdoi2016]生成魔咒

●BZOJ 4516 [Sdoi2016]生成魔咒

ati tle mes http sin aps spa 代碼 需要

題鏈:

http://www.lydsy.com/JudgeOnline/problem.php?id=4516

題解:

把串反過來後,問題變為求每個後綴的互不相同的子串個數。
首先用倍增算法求出 sa[],rank[],height[],然後對 height[]數組建立 ST表。
接著求出整個串的子串個數,ans+=N-sa[i]-height[i]。(我從0開始編號的)
式子的含義就是考慮每個後綴相比它的前一名,多了幾個與之前不同的且串頭為該後綴的頭的子串。
(一定要清晰地懂得並理解那個式子哦)

之前得出了0 位置開始的後綴(即整個串)的子串個數,
那麽現在就需要把 rank[0]這個後綴從排好序的後綴數組中去除


然後維護出新的後綴(即從1位置開始的後綴)的子串個數
怎麽做呢,反向考慮 ans的求法:
把rank[0]產生的貢獻減去(包括和它上面一名以及和它下面一名產生的貢獻),相當於該後綴被去除了。
這時排在rank[0]上面一位的後綴(設為 u),和排在rank[0]下面一位的後綴(設為 d),
挨在了一起,那麽要加上 u 後綴和 d 後綴的貢獻
然後就得到了新的後綴的子串個數。
之後的其它後綴的計算就類似了。

另外再提一下,在找當前後綴的上一名後綴和下一名後綴時,找到的必須是還在後綴數組中(即還沒有被去除),
可以用類似並查集的思想維護(好吧,是路徑壓縮的思想),做到均攤 O(1)。

除開倍增算法和求ST表的復雜度 O(N)

代碼:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 100500
#define filein(x) freopen(#x".in","r",stdin);
#define fileout(x) freopen(#x".out","w",stdout);
using namespace std;
int sa[MAXN],rak[MAXN],hei[MAXN];
int up[MAXN],down[MAXN],A[MAXN],log2[MAXN],stm[MAXN][20];
bool vis[MAXN];
void build(int N,int M){
	static int cc[MAXN],ta[MAXN],tb[MAXN],*x,*y,h,p;
	x=ta; y=tb; h=0; A[N]=-1;
	for(int i=0;i<M;i++) cc[i]=0;
	for(int i=0;i<N;i++) cc[x[i]=A[i]]++;
	for(int i=1;i<M;i++) cc[i]+=cc[i-1];
	for(int i=N-1;i>=0;i--) sa[--cc[x[i]]]=i;
	for(int k=1;p=0,k<N;k<<=1){
		for(int i=N-k;i<N;i++) y[p++]=i;
		for(int i=0;i<N;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
		for(int i=0;i<M;i++) cc[i]=0;
		for(int i=0;i<N;i++) cc[x[y[i]]]++;
		for(int i=1;i<M;i++) cc[i]+=cc[i-1];
		for(int i=N-1;i>=0;i--) sa[--cc[x[y[i]]]]=y[i];
		swap(x,y); y[N]=-1; x[sa[0]]=0; M=1;
		for(int i=1;i<N;i++)
			x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?M-1:M++;
		if(M>=N) break;
	}
	for(int i=0;i<N;i++) rak[sa[i]]=i;
	for(int i=0,j;i<N;i++){
		if(h) h--;
		if(rak[i]){
			j=sa[rak[i]-1];
			while(A[i+h]==A[j+h]) h++;
		}
		stm[rak[i]][0]=hei[rak[i]]=h;
	}
	for(int k=1;k<=log2[N];k++)
        for(int i=(1<<k)-1;i<N;i++)
            stm[i][k]=min(stm[i-(1<<(k-1))][k-1],stm[i][k-1]);
}
int query(int l,int r,int N){
    static int k;
    if(l==-1||r==-1||l==N||r==N) return 0;
    if(l>r) swap(l,r); l++;
    k=log2[r-l+1];
    return min(stm[l+(1<<k)-1][k],stm[r][k]);
}
int find(int i,int *to,const int &N){
	if(i==-1||i==N||!vis[i]) return i;
	return to[i]=find(to[i],to,N);
}
void solve(int N){
	static long long now,ANS[MAXN];
	for(int i=1;i<N;i++) up[i]=i-1,down[i-1]=i; up[0]=-1; down[N-1]=N;
	for(int i=0;i<N;i++) now+=1ll*N-sa[i]-hei[i]; 
	ANS[N]=now; sa[N]=N;
	for(int i=0,r,u,d;i<N-1;i++){
		r=rak[i]; vis[r]=1; u=find(r,up,N); d=find(r,down,N);
		now-=1ll*N-sa[r]-query(u,r,N);
		now-=1ll*N-sa[d]-query(d,r,N);
		now+=1ll*N-sa[d]-query(u,d,N);
		ANS[N-i-1]=now;
	}
	for(int i=1;i<=N;i++) printf("%lld\n",ANS[i]);
}
int main()
{
	filein(incantation);fileout(incantation);
	static int tmp[MAXN];
	log2[1]=0; for(int i=2;i<=100000;i++) log2[i]=log2[i>>1]+1;
	int N,cnt; scanf("%d",&N);
	for(int i=N-1;i>=0;i--) scanf("%d",&A[i]),tmp[i]=A[i];
	sort(tmp,tmp+N); 
	cnt=unique(tmp,tmp+N)-tmp;
	for(int i=0;i<N;i++) A[i]=lower_bound(tmp,tmp+cnt,A[i])-tmp;
	build(N,N);
	solve(N);
	return 0;
}

●BZOJ 4516 [Sdoi2016]生成魔咒