接着上一篇讲,这个最长不下降子序列,先给大家看一下题目。
给定一个长度为 n 的序列 A(n ≤ 5000),求出一个最长的 A 的子序列,满足该子序列的后一个 元素不小于前一个元素。
算法一
设 f(i) 表示以 Ai 为结尾的最长不下降子序列的长度,则所求为 max1≤i≤n f(i)。
计算 f(i) 时,尝试将 Ai 接到其他的最长不下降子序列后面,以更新答案。于是可 以写出这样的状态转移方程:f(i) = max(f(i),f(j)+1)
容易发现该算法的时间复杂度为 O(n²)。
int a[MAXN], d[MAXN];
int dp() {
d[1] = 1;
int ans = 1;
for (int i = 2; i <= n; i++) {
d[i] = 1;
for (int j = 1; j < i; j++)
if (a[j] <= a[i]) {
d[i] = max(d[i], d[j] + 1);
ans = max(ans, d[i]);
}
}
return ans;
}
算法二
当 n 的范围扩大到 n ≤ 10⁵ 时,第一种做法就不够快了,下面给出了一个 O(n log n) 的做法。
首先,定义 a1…an 为原始序列,d 为当前的不下降子序列,len 为子序列的长 度,那么 dlen 就是长度为 len 的不下降子序列末尾元素。
初始化:d1 = a1, len = 1。
现在我们已知最长的不下降子序列长度为 1,那么我们让 i 从 2 到 n 循环,依次求 出前 i 个元素的最长不下降子序列的长度,循环的时候我们只需要维护好 d 这个 数组还有 len 就可以了。关键在于如何维护。
考虑进来一个元素 ai:
1. 元素大于等于 dlen,直接将该元素插入到 d 序列的末尾。
2. 元素小于 dlen,找到 第一个 大于它的元素,插入进去,丢弃在它之后的全部 元素。
for (int i = 0; i < n; ++i) scanf("%d", a + i);
memset(dp, 0x1f, sizeof dp);
mx = dp[0];
for (int i = 0; i < n; ++i) {
*std::upper_bound(dp, dp + n, a[i]) = a[i];
}
ans = 0;
while (dp[ans] != mx) ++ans;
以上为优化后的最长不下降子序列代码,大概用了两篇博客来写动态规划,这只是基础,后面还会有很多不同类型的动态规划,请看后面几篇,稍后更新。