题意:给你两个数组,让你将这两个数组合并成一个数组,每个数组的内部顺序不能改变,让你求合并能得到字典序最大的数组。
(n<=1e5, 1 <= a[i],b[i] <= 10)
思路:正常的贪心的思路是用两个指针从两个开头开始比较,谁大谁先放,但是问题在于碰到两个相等的时候谁先放,思路是往后
找第一个不相等的,谁的大谁先,但是这样会n^2复杂度。我们可以利用后缀数组将两个数组合并(中间位置要放0,比如 1 1 X 1 1
2比较后应该先放后面的1),预处理出以每个位置开头的rank,这样遇到相等的时候直接比较下rank就知道哪个应该先放了。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e5+5;
int t1[maxn], t2[maxn], c[maxn];
int ra[maxn], height[maxn];
int sa[maxn], str[maxn];
bool cmp(int *r, int a, int b, int l)
{
return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int str[], int sa[], int ra[], int height[], int n, int m)
{
n++;
int i, j, p, *x = t1, *y = t2;
for(i = 0; i < m; i++) c[i] = 0;
for(i = 0; i < n; i++) c[x[i]=str[i]]++;
for(i = 1; i < m; i++) c[i] += c[i-1];
for(i = n-1; i >= 0; i--) sa[--c[x[i]]] = i;
for(j = 1; j <= n; j<<=1)
{
p = 0;
for(i = n-j; i < n; i++) y[p++] = i;
for(i = 0; i < n; i++) if(sa[i] >= j) y[p++] = sa[i]-j;
for(i = 0; i < m; i++) c[i] = 0;
for(i = 0; i < n; i++) c[x[y[i]]]++;
for(i = 1; i < m; i++) c[i] += c[i-1];
for(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; i < n; i++)
x[sa[i]] = cmp(y, sa[i-1], sa[i], j) ? p-1 : p++;
if(p >= n) break;
m = p;
}
int k = 0;
n--;
for(i = 0; i <= n; i++) ra[sa[i]] = i;
for(i = 0; i < n; i++)
{
if(k) k--;
j = sa[ra[i]-1];
while(str[i+k]==str[j+k]) k++;
height[ra[i]] = k;
}
}
int ans[maxn];
int main(void)
{
int t, ca = 1;
cin >> t;
while(t--)
{
int n, m;
scanf("%d%d", &n, &m);
int len = n+m+1;
for(int i = 0; i < n; i++)
scanf("%d", &str[i]);
str[n] = 0;
for(int i = 0; i < m; i++)
scanf("%d", &str[i+n+1]);
str[len] = 0; //我的破板子必须要加这个不然wa
da(str, sa, ra, height, len, 127);
int cnt = 0, p1 = 0, p2 = n+1;
while(p1 < n && p2 < len)
{
if(str[p1] > str[p2]) ans[cnt++] = str[p1++];
else if(str[p1] < str[p2]) ans[cnt++] = str[p2++];
else if(ra[p1] > ra[p2]) ans[cnt++] = str[p1++];
else ans[cnt++] = str[p2++];
}
while(p1 < n) ans[cnt++] = str[p1++];
while(p2 < len) ans[cnt++] = str[p2++];
printf("Case %d: ", ca++);
for(int i = 0; i < cnt; i++)
printf("%d", ans[i]);
puts("");
}
return 0;
}