BZOJ 4278: [ONTAK2015]Tasowanie 后缀数组

4278: [ONTAK2015]Tasowanie

Time Limit: 10 Sec  Memory Limit: 256 MB
Submit: 426  Solved: 199
[Submit][Status][Discuss]

Description

给定两个数字串A和B，通过将A和B进行二路归并得到一个新的数字串T，请找到字典序最小的T。

Input

第一行包含一个正整数n(1<=n<=200000)，表示A串的长度。

第二行包含n个正整数，其中第i个数表示A[i](1<=A[i]<=1000)。

第三行包含一个正整数m(1<=m<=200000)，表示B串的长度。

第四行包含m个正整数，其中第i个数表示B[i](1<=B[i]<=1000)。

Output

输出一行，包含n+m个正整数，即字典序最小的T串。

Sample Input

6
1 2 3 1 2 4
7
1 2 2 1 3 4 3

Sample Output

1 1 2 2 1 2 3 1 2 3 4 3 4

在这个清闲的早晨，感觉终于完全理解了后缀数组

看到战友们这两天都做了这题，决定写一写

想了O(n)的贪心。。。然后GG了

无奈之下翻题解，WA后缀数组

两个串接上中间用INF分开 求一下rank就OK了

这题竟然要求行尾必须有空格。。。PE一发

#include<cmath>#include<ctime>#include<cstdio>#include<cstring>#include<cstdlib>#include<complex>#include<iostream>#include<algorithm>#include<iomanip>#include<vector>#include<string>#include<bitset>#include<queue>#include<set>#include<map>using namespace std;typedef long long ll;inline int read(){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}inline void print(int x){if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}const int N=400100;int buc[N],sa1[N],sa2[N],rk1[N],rk2[N],ht[N];int *sa=sa1,*tp=sa2,*rk=rk1,*tmp=rk2;int s[N];int n,m;void get_ht(){register int i,j,k=0;for(i=1;i<n;++i){j=rk[i]-1;while(s[sa[rk[i]]+k]==s[sa[j]+k])k++;ht[rk[i]]=k;if(k)k--;}}void get_sa(int len){register int i,k,j;register bool flag(0);for(i=1;i<=len;++i)buc[s[i]]++;for(i=1;i<=1010;++i)buc[i]+=buc[i-1];for(i=len;i;--i)sa[buc[s[i]]--]=i;for(i=1;i<=len;++i)rk[sa[i]]=rk[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]);for(k=1;k<=len;k<<=1){for(i=1;i<=len;++i)buc[rk[sa[i]]]=i;for(i=len;i;--i)if(sa[i]>k)tp[buc[rk[sa[i]-k]]--]=sa[i]-k;for(i=len-k+1;i<=len;++i)tp[buc[rk[i]]--]=i;for(i=1;i<=len;++i){tmp[tp[i]]=tmp[tp[i-1]]+(rk[tp[i]]!=rk[tp[i-1]]||rk[tp[i]+k]!=rk[tp[i-1]+k]);if(tmp[tp[i]]==len)break;}swap(tmp,rk);swap(sa,tp);if(flag)break;}}void solve(){register int k,i(1),j(n+2);for(k=1;k<=n+m;k++){rk[i]<rk[j]?(print(s[i]),i++):(print(s[j]),j++);putchar(' ');}puts("");//rk[i]<rk[j]?print(s[i]):print(s[j]);}int main(){register int i;n=read();for(i=1;i<=n;++i)s[i]=read();m=read();for(i=n+2;i<=m+n+1;++i)s[i]=read();s[n+1]=s[n+m+2]=1010;get_sa(n+m+2);solve();return 0;} /*61 2 3 1 2 471 2 2 1 3 4 31 1 2 2 1 2 3 1 2 3 4 3 4*/