bzoj 4278: [ONTAK2015]Tasowanie （后缀数组）

4278: [ONTAK2015]Tasowanie

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 313   Solved: 140
[ 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

HINT

Source

By Claris

[ Submit][ Status][ Discuss]

题解：后缀数组

将两个串用一个分隔符连接起来，然后求rank数组.需要注意的是分隔符一定要加较大的，否则会出现2 0 2 1(0是分隔符)正确的答案是212，但是会选择出221.在整个串的最后也要加一个大的分隔符，否则会出现2 1 1003 2正确的答案还是212，但是如果不加分隔符会选择出221.

不过这题的数据较水，网上的很多程序都是有bug的，但是都过了。。。。表示还是应该加强一下数据。

将两个头指针放到两个串开始的位置，每次选取开头位置中rank较小的。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 5000007
using namespace std;
int n,m,a[N],v[N],xx[N],yy[N],*x,*y,sa[N],rank[N],p;
int cmp(int i,int j,int l) {
	return y[i]==y[j]&&(i+l>n?-1:y[i+l])==(j+l>n?-1:y[j+l]);
}
void get_sa()
{
	x=xx; y=yy;
	memset(v,0,sizeof(v));
	int m1=1005;
	for (int i=1;i<=n;i++) v[x[i]=a[i]]++;
	for (int i=1;i<=m1;i++) v[i]+=v[i-1];
	for (int i=n;i>=1;i--) sa[v[x[i]]--]=i;
	for (int k=1;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<=m1;i++) v[i]=0;
		for (int i=1;i<=n;i++) v[x[y[i]]]++;
		for (int i=1;i<=m1;i++) v[i]+=v[i-1];
		for (int i=n;i>=1;i--) sa[v[x[y[i]]]--]=y[i];
		swap(x,y); p=2; x[sa[1]]=1;
		for (int i=2;i<=n;i++) 
		 x[sa[i]]=cmp(sa[i],sa[i-1],k)?p-1:p++;
		if (p>n) break;
		m1=p+1;
	}
	for (int i=1;i<=n;i++) rank[sa[i]]=i;
}
int main()
{
	freopen("a.in","r",stdin);
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	a[++n]=1003;
	scanf("%d",&m);
	for (int i=1;i<=m;i++) scanf("%d",&a[++n]);
	a[++n]=1003;
	get_sa();
	int l=1; int r=n-m;
//	cout<<l<<" "<<r<<endl;
    int cnt=0;
    for (int i=1;i<=n-2;i++){
    	if (rank[l]<rank[r]) printf("%d ",a[l]),l++;
		else printf("%d ",a[r]),r++;
		if (l==n-m-1) {
			for (int j=r;j<=n-1;j++) printf("%d ",a[j]);
			break;
		}
		if (r==n) {
			for (int j=l;j<=n-m-2;j++) printf("%d ",a[j]);
			break;
		}
	}
	printf("\n");
}