[bzoj4278][ONTAK2015]Tasowanie_后缀数组_贪心

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;
}

小结：注意后缀数组的本质。