bzoj 4516: [Sdoi2016]生成魔咒 字尾陣列
阿新 • • 發佈:2019-02-10
這道題目最簡單粗暴的方法是建立sam,然後每個點用一個map儲存,統計答案就用len[i]-len[fa[i]]即可。。
考慮字尾陣列。將整個陣列反過來,那麼每次就相當於查詢字尾[i,n]中有多少本質不同的子串。那麼就可以建出字尾陣列,然後新加入一個字尾[p],那麼首先需要得到這個字尾中有多少子串是重複的,相當於求這個字尾與字尾集合[p+1,n]的最長lcp。由於已經得到了字尾陣列,只需要知道字尾[p]在[p+1,n]中相鄰的兩個字尾(樹狀陣列維護),然後利用rmq求任意字尾的lcp即可。
注意開long long。
AC程式碼如下:
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #define N 200005 using namespace std; int n,bin[25],lg2[N],f[17][N],num[N],sa[N],rnk[N],q[N],hgt[N],sum[N],c[2][N]; struct node{ int x,id; }a[N]; bool cmp(node u,node v){ return u.x<v.x; } int read(){ int x=0; char ch=getchar(); while (ch<'0' || ch>'9') ch=getchar(); while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); } return x; } void solve_sa(){ sort(a+1,a+n+1,cmp); int i,k,cnt=0; for (i=1; i<=n; i++){ if (i==1 || a[i].x!=a[i-1].x) cnt++; rnk[a[i].id]=num[a[i].id]=cnt; } for (i=1; i<=n; i++) sum[rnk[i]]++; for (i=2; i<=cnt; i++) sum[i]+=sum[i-1]; for (i=n; i; i--) sa[sum[rnk[i]]--]=i; for (k=1; cnt<n; k<<=1){ for (i=1; i<=k; i++) q[i]=n-k+i; memset(sum,0,sizeof(sum[0])*(cnt+1)); for (i=1; i<=n-k; i++) sum[rnk[i+k]]++; sum[0]=k; for (i=1; i<=cnt; i++) sum[i]+=sum[i-1]; for (i=n-k; i; i--) q[sum[rnk[i+k]]--]=i; memset(sum,0,sizeof(sum[0])*(cnt+1)); for (i=1; i<=n; i++) sum[rnk[i]]++; for (i=2; i<=cnt; i++) sum[i]+=sum[i-1]; for (i=n; i; i--) sa[sum[rnk[q[i]]]--]=q[i]; cnt=0; memcpy(q,rnk,sizeof(rnk[0])*(n+1)); for (i=1; i<=n; i++){ if (i==1 || q[sa[i]]!=q[sa[i-1]] || q[sa[i]+k]!=q[sa[i-1]+k]) cnt++; rnk[sa[i]]=cnt; } } } void solve_hgt(){ int i,j,k=0; for (i=1; i<=n; i++){ if (k) k--; j=sa[rnk[i]-1]; while (num[j+k]==num[i+k]) k++; hgt[rnk[i]]=k; } } int lcp(int x,int y){ if (!x || x>n) return 0; if (x>y) swap(x,y); x++; int k=lg2[y-x+1]; return min(f[k][x],f[k][y-bin[k]+1]); } void ins(int k,int x){ int i; for (i=x; i<=n; i+=i&-i) c[k][i]=max(c[k][i],x); } int find(int k,int x){ int t=0; for (; x; x^=x&-x) t=max(t,c[k][x]); return t; } int main(){ n=read(); int i,j; for (i=n; i; i--){ a[i].x=read(); a[i].id=i; } solve_sa(); solve_hgt(); memcpy(f[0],hgt,sizeof(hgt)); lg2[1]=0; bin[0]=1; bin[1]=2; for (i=1; i<=16; i++){ bin[i+1]=bin[i]<<1; for (j=bin[i]; j<bin[i+1]; j++) lg2[j]=i; for (j=1; j<=n; j++){ f[i][j]=f[i-1][j]; if (j+bin[i-1]<=n) f[i][j]=min(f[i][j],f[i-1][j+bin[i-1]]); } } int t1,t2; long long ans=0; for (i=n; i; i--){ ans+=n-i+1; t1=find(0,rnk[i]-1); t2=n-find(1,n-rnk[i])+1; ans-=max(lcp(t1,rnk[i]),lcp(t2,rnk[i])); printf("%lld\n",ans); ins(0,rnk[i]); ins(1,n-rnk[i]+1); } return 0; }
by lych
29016.4.19