bzoj 4278 [ONTAK2015]Tasowanie（SA，贪心）

【题意】

    给定两个字符串，求二路归并后最小字典序的字符串。

【思路】

    连接两个字符串后求出rank数组。通过比较rank数组进行二路归并。

【代码】

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define FOR(a,b,c) for(int a=b;a<=c;a++)
using namespace std;

typedef long long ll;
const int N = 4e3+10;

ll read()
{
    char c=getchar();
    ll f=1,x=0;
    while(!isdigit(c)) {
        if(c=='-') f=-1;
        c=getchar();
    }
    while(isdigit(c))
        x=x*10+c-'0', 
        c=getchar();
    return x*f;
}

int s[N];
int t[N],t2[N],c[N],sa[N],rank[N];

void build_sa(int m,int n)
{
    int *x=t,*y=t2;
    FOR(i,0,m-1) c[i]=0;
    FOR(i,0,n-1) c[x[i]=s[i]]++;
    FOR(i,0,m-1) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1;k<=n;k<<=1) {
        int p=0;
        FOR(i,n-k,n-1) y[p++]=i;
        FOR(i,0,n-1) if(sa[i]>=k) y[p++]=sa[i]-k;
        
        FOR(i,0,m-1) c[i]=0;
        FOR(i,0,n-1) c[x[y[i]]]++;
        FOR(i,0,m-1) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
         
        swap(x,y);
        p=1; x[sa[0]]=0;
        FOR(i,1,n-1)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p>=n) break;
        m=p;
    }
}
void get_rank(int n) 
{
    FOR(i,0,n-1) rank[sa[i]]=i;
}

int n,m,len,a[N],b[N],ans[N];

int main()
{
    n=read();
    FOR(i,0,n-1) a[i]=read(),s[len++]=a[i];
    s[len++]=1001;
    m=read();
    FOR(i,0,m-1) b[i]=read(),s[len++]=b[i];
    s[len++]=1001;
    
    build_sa(1003,len);
    get_rank(len);
    
    int p1=0,p2=0,tot=0;
    while(p1<n || p2<m) {
        if(p1>=n) printf("%d ",b[p2++]);
        else if(p2>=m) printf("%d ",a[p1++]);
        else {
            if(rank[p1]<rank[n+1+p2]) printf("%d ",a[p1++]);
            else printf("%d ",b[p2++]);
        }
    }
    return 0;
}