BZOJ 4278: [ONTAK2015]Tasowanie

Description

将两个字符串合并成一个，让字典序最小。\(n\leqslant 2\times 10^5\)

Solution

后缀数组+贪心。

和起来排个序，然后按\(rank\)贪心选就好，注意在字符串最后放个比较大的分割符。

Code

/**************************************************************
    Problem: 4278
    User: BeiYu
    Language: C++
    Result: Accepted
    Time:2012 ms
    Memory:14964 kb
****************************************************************/
 
#include <bits/stdc++.h>
using namespace std;
 
const int N = 500050;
 
inline int in(int x=0,char ch=getchar()) { for(;ch>'9'||ch<'0';ch=getchar());
    for(;ch>='0'&&ch<='9';x=x*10+ch-'0',ch=getchar());return x; }
 
int n,m=1001;
int a[N];
 
namespace SA {
    int t1[N],t2[N],c[N],sa[N],rk[N],ht[N];
     
    void get_sa(int a[],int n=::n,int m=::m) {
        int *x=t1,*y=t2;
        for(int i=1;i<=m;i++) c[i]=0;
        for(int i=1;i<=n;i++) c[x[i]=a[i]]++;
        for(int i=1;i<=m;i++) c[i]+=c[i-1];
        for(int i=n;i;--i) sa[c[x[i]]--]=i;
        for(int k=1,p=0;k<n;k<<=1,p=0) {
            for(int i=n-k+1;i<=n;i++) y[++p]=i;
            for(int i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;
            for(int i=1;i<=m;i++) c[i]=0;
            for(int i=1;i<=n;i++) c[x[i]]++;
            for(int i=1;i<=m;i++) c[i]+=c[i-1];
            for(int i=n;i;--i) sa[c[x[y[i]]]--]=y[i];
            swap(x,y),x[sa[1]]=p=1;
            for(int i=2;i<=n;i++)
                x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?p:++p;
            if(p>=n) break;m=p;
        }
    }
    void get_ht(int a[],int n=::n) {
        for(int i=1;i<=n;i++) rk[sa[i]]=i;
        for(int i=1,j,k=0;i<=n;ht[rk[i++]]=k) 
            for(j=sa[rk[i]-1],k=k?k-1:k;a[i+k]==a[j+k];k++);
    }
};
 
int main() {
    int l1=in();
    for(int i=1;i<=l1;i++) a[++n]=in();
    int l2=in();
    a[++n]=1000;
    for(int i=1;i<=l2;i++) a[++n]=in();
    a[++n]=1000;
    SA::get_sa(a,n,m);
    SA::get_ht(a,n);
//  for(int i=1;i<=n;i++) cout<<SA::sa[i]<<" ";cout<<endl;
//  for(int i=1;i<=n;i++) cout<<SA::ht[i]<<" ";cout<<endl;
    for(int i=1,p1=1,p2=1;i<=l1+l2;i++) {
        if(p1>l1) printf("%d ",a[l1+1+p2]),p2++;
        else if(p2>l2) printf("%d ",a[p1]),p1++;
        else {
            if(SA::rk[p1]>SA::rk[l1+1+p2]) printf("%d ",a[l1+1+p2]),p2++;
            else printf("%d ",a[p1]),p1++;
        }
    }return 0;
}