BZOJ4278 [ONTAK2015]Tasowanie[后缀数组+贪心]

题目

求两数组归并后的数组最小字典序排列。

嘛，可能本人在贪心这块还是太弱了（或者说什么都弱），如果不知道是字符串题估计也想不起来用sa。

显然看得出归并时字典序小的那个数组先往里面加，这就是要比较两数组后缀的rank，方法就把两串相拼做后缀排序后比较。

这里附下贪心正确性证明，反正我不太会，只是感性认识一下。

Upd:突然想起来这题其实可以hash+二分比较，码量稍小，不想写了

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
template<typename T>inline char MIN(T&A,T B){return A<B?A=B,1:0;}
template<typename T>inline char MAX(T&A,T B){return A>B?A=B,1:0;}
template<typename T>inline T _min(T A,T B){return A<B?A:B;}
template<typename T>inline T _max(T A,T B){return A>B?A:B;}
template<typename T>inline T read(T&x){
    x=0;int f=0;char c;while(!isdigit(c=getchar()))if(c=='-')f=1;
    while(isdigit(c))x=(x<<1)+(x<<3)+(c^48),c=getchar();return f?x=-x:x;
}
const int N=200000+7;
int A[N<<1],a[N],b[N],l,m,n;

int rk[N<<1],sa[N<<1],cnt[N<<1],y[N<<1],t=1001,p;
inline void suffix_sort(){
    for(register int i=1;i<=l;++i)++cnt[rk[i]=A[i]];
    for(register int i=1;i<=t;++i)cnt[i]+=cnt[i-1];
    for(register int i=l;i;--i)sa[cnt[A[i]]--]=i;
    for(register int k=1;k<l;k<<=1,p=0){
        for(register int i=l-k+1;i<=l;++i)y[++p]=i;
        for(register int i=1;i<=l;++i)if(sa[i]>k)y[++p]=sa[i]-k;
        for(register int i=1;i<=t;++i)cnt[i]=0;
        for(register int i=1;i<=l;++i)++cnt[rk[y[i]]];
        for(register int i=1;i<=t;++i)cnt[i]+=cnt[i-1];
        for(register int i=l;i;--i)sa[cnt[rk[y[i]]]--]=y[i];
        swap(rk,y);rk[sa[1]]=p=1;
        for(register int i=2;i<=l;++i)rk[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i-1]+k]==y[sa[i]+k]?p:++p;
        if(p==l)break;t=p;
    }//for(register int i=1;i<=l;++i)printf("%d %d\n",i,rk[i]);
}
//
int main(){//freopen("tmp.in","r",stdin);freopen("tmp.out","w",stdout);
    read(n);for(register int i=1;i<=n;++i)A[i]=read(a[i]);A[n+1]=1001;
    read(m);for(register int i=1;i<=m;++i)A[n+i+1]=read(b[i]);A[n+m+2]=1001;
    l=n+m+2;suffix_sort();int i=1,j=1;
    while(i<=n||j<=m){
        if(rk[i]<rk[n+j+1])printf("%d ",a[i++]);
        else printf("%d ",b[j++]);
    }puts("");
    return 0;
}