El Psy Congroo!
题意
给出一个长为(n)的数组(a),与一个长为(m)的数组(b)
求数组(a)与数组(b)的最长公共上升子序列长度并输出任意一组方案
(1leq n,mleq 5000)
解法
这题很有意思
首先对于这类匹配问题,我们一般可以设(f[i][j])表示(a)数组中考虑到(a_i)且(a_i)必须选,(b)数组中考虑到(j)的最长公共上升子序列的长度
设出方程以后转移就比较好想了
当(a[i]=a[j])时
(f[i][j]=max{f[k][j-1]}+1),其中(k<i)且(a[k]<a[i])
当(a[i] eq a[j])时
(f[i][j]=f[i][j-1])
想到转移以后,显然的思路是直接按照转移方程写,即第一层循环枚举(i from 1...n),第二层循环枚举(j from 1...m)
但这样我们会发现(max)的条件不好维护
朴素的转移是(O(n^3))的,就是用线段树优化转移也是(O(n^2 log n))的,对于(n leq 5000)的数据来说还是有点卡,更别说线段树自带的大常数了
考虑转换一下枚举顺序,即先枚举(j from 1...m),再枚举(i from 1...n)
这样我们只需要在枚举的过程中顺便记录一下(max)就可以在之后(O(1))进行转移了
但是有一个问题,之前我们需要的条件判断是(k<i)且(a[k]<a[i])
第一个条件还好办,主要是第二个条件:对于(i)不确定的情况下,我们如何确定(a[i])的值呢?
我们会发现,由于是公共子序列,又由于(a[i]=b[j])的时候才能产生贡献,我们把(a[i])替换为(b[j])就好了
至于求方案,记录一下更新状态时是由哪个状态转移过来的就行了
代码
#include <cstdio>
#include <algorithm>
using namespace std;
#define Pii pair<int, int>
#define x first
#define y second
const int N = 5e3 + 10;
int n, m;
int a[N], b[N], f[N][N];
Pii lst[N][N];
void output(int i, int j) {
if (!i) return;
output(lst[i][j].x, lst[i][j].y);
printf("%d ", a[i]);
}
int main() {
// freopen("okarin.in", "r", stdin);
// freopen("okarin.out", "w", stdout);
scanf("%d", &n);
for (int i = 1; i <= n; ++i) scanf("%d", a + i);
scanf("%d", &m);
for (int i = 1; i <= m; ++i) scanf("%d", b + i);
for (int j = 1; j <= m; ++j) {
int max = 0;
Pii pre = make_pair(0, 0);
for (int i = 1; i <= n; ++i) {
if (a[i] == b[j]) {
f[i][j] = max + 1;
lst[i][j] = pre;
} else {
if (a[i] < b[j] && f[i][j - 1] > max)
max = f[i][j - 1], pre = make_pair(i, j - 1);
f[i][j] = f[i][j - 1];
lst[i][j] = lst[i][j - 1];
}
}
}
int ans = 0, pos = 0;
for (int i = 1; i <= n; ++i)
if (f[i][m] > ans)
ans = f[i][m], pos = i;
printf("%d
", ans);
output(pos, m);
return 0;
}