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;
} 
/*
6
1 2 3 1 2 4
7
1 2 2 1 3 4 3

1 1 2 2 1 2 3 1 2 3 4 3 4
*/