[bzoj4278][ONTAK2015]Tasowanie_字尾陣列_貪心
阿新 • • 發佈:2018-12-12
Tasowanie bzoj-4278 ONTAK-2015
題目大意:給定兩個字串,求兩個字串二路歸併之後生成的字典序最小的字串是什麼。
註釋:$1\le len_1,len_2\le 2\cdot 10^5$。
想法:
和佇列變換比較像啊。
我們將兩個串接在一起然後建立字尾陣列即可。
緊接著像那道題一樣,在$sa$上弄兩個指標貪心地移動即可。
Code:
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; #define N 400050 int Ws[N],wv[N],wa[N],wb[N],r[N],n,rk[N],ht[N],sa[N],m,la,lb; int a[N],b[N]; void build_sa() { m=1002; int i,j,p,*x=wa,*y=wb,*t; for(i=0;i<m;i++) Ws[i]=0; for(i=0;i<n;i++) Ws[x[i]=r[i]]++; for(i=1;i<m;i++) Ws[i]+=Ws[i-1]; for(i=n-1;i>=0;i--) sa[--Ws[x[i]]]=i; for(p=j=1;p<n;j<<=1,m=p) { for(p=0,i=n-j;i<n;i++) y[p++]=i; for(i=0;i<n;i++) if(sa[i]-j>=0) y[p++]=sa[i]-j; for(i=0;i<n;i++) wv[i]=x[y[i]]; for(i=0;i<m;i++) Ws[i]=0; for(i=0;i<n;i++) Ws[wv[i]]++; for(i=1;i<m;i++) Ws[i]+=Ws[i-1]; for(i=n-1;i>=0;i--) sa[--Ws[wv[i]]]=y[i]; for(t=x,x=y,y=t,i=p=1,x[sa[0]]=0;i<n;i++) { if(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]) x[sa[i]]=p-1; else x[sa[i]]=p++; } } for(i=1;i<n;i++) rk[sa[i]]=i; for(i=p=0;i<n-1;ht[rk[i++]]=p) for(p?p--:0,j=sa[rk[i]-1];r[i+p]==r[j+p];p++); } int main() { scanf("%d",&la); int i; for(i=0;i<la;i++) { scanf("%d",&a[i]); r[n++]=a[i]; } scanf("%d",&lb); r[n++]=1001; for(i=0;i<lb;i++) { scanf("%d",&b[i]); r[n++]=b[i]; } r[n++]=0; build_sa(); int j=0,k=la+1; while(j<la&&k<n-1) { if(rk[j]<rk[k]) printf("%d ",r[j++]); else printf("%d ",r[k++]); } while(j<la) printf("%d ",r[j++]); while(k<n-1) printf("%d ",r[k++]); puts(""); return 0; }
小結:注意字尾陣列的本質。