来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/longest-increasing-subsequence
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一、题目描述
给定一个无序的整数数组,找到其中最长上升子序列的长度。
示例:
- 输入: [10,9,2,5,3,7,101,18]
- 输出: 4
- 解释: 最长的上升子序列是 [2,3,7,101],它的长度是 4。
说明:
- 可能会有多种最长上升子序列的组合,你只需要输出对应的长度即可。
- 你算法的时间复杂度应该为O(n^2)。
进阶:你能将算法的时间复杂度降低到O(nlogn)吗?
二、题解
首先要搞清楚的是题意:求最长上升子序列。子序列不是子串,两者的区别是子序列可以不连续,子串必须连续。
例如:[1, 3, 5, 2, 9],最长上升子序列是[1, 3, 5, 9],最长上升子串是[1, 3, 5]。
2.1 暴力法
题目的一个要求是用O(n^2)的时间复杂度来完成这个问题,如果有O(n^2)的时间复杂度,可以直接使用暴力破解的办法:
1 2 3 4 5 6 7 8 |
// 伪代码 for (i = 0; i < size; i++) { for (j = i; j < size; j++) { // 存在一个上升序列,和加一 if (nums[j] >= nums[i]) val[i][j] +=1; } } |
算法的时间复杂度和空间复杂度都是O(n^2)。
2.2 动态规划
以dp[i]
表示到第i个元素时的最大上升子序列长度,那么dp[i]就等于数组中上一个小于nums[i]的元素的最大上升子序列长度加1。这句话看起来比较难理解,可以通过下面的例子来理解。
以[10,9,2,5,3,7]为例,i=5时如何求dp[i]?此时nums[i]=7,要想构造上升序列,它前面的数字都小于7才行。左边小于7的数字有三个:2、5和3。
- 在数字5的地方,最大上升子序列的长度是2,上升序列是[2,5];
因为7大于3个数字,所以到达7时,可以构造出来的上升序列有:[2,7], [2,5,7], [2,3,7],这个时候最大的上升序列长度是3,3是从前面所有子序列的最大长度加1得到的。所以可以归纳出来的状态转移方程为:
$$dp[i]=max(dp[0], dp[1],..., dp[i-1])+1, n<i, nums[n]<=nums[i]. $$ \( \)
使用动态规划的时间复杂度也是O(n^2),空间复杂度O(n)。
2.3 二分法求解
二分法不容易想出来,即使是看题解也是看了几遍才明白。
最大上升子序列长度为i
以[10, 9, 2, 5, 7]为例解释tails数组每个元素的意义:
- 长度为1的子序列有5个,每个元素都可以看作独立的子序列,其中最小的数字是2,所以tails[1]=2。
- 长度为2的子序列有[2, 5], [2, 7]和[5, 7],其中末尾数最小的是5,所以tails[2]=5。
- 长度为3的子序列只有[2, 5, 7],末尾数只有一个就是7,所以tails[3]=7。
tails数组的算法:
- 在数组中找到第一个大于nums[i]的元素,替换它。
- 如果找不到一个大于nums[i]的元素,插入到数组最后。
如何找到第一个大于nums[i]的元素?因为序列是有序的,所以最快的算法是二分查找,只要找到第一个大于等于它的值就可以了。时间复杂度为O(nlogn),空间复杂度为O(n)。
为什么一定要保存每个序列长度的最小值
一个要注意的问题是tails[i]表示的是长度为i的子序列的末尾元素的最小值,以[1, 5, 3, 4]为例,最大上升子序列长度为2的有两个:[1, 5]和[1, 3],如果不取最小的末尾元素3,那么tails数组就是[1, 5],当遍历到4的时候,找到第一个大于它的数是5,此时下标等于2,也就是说最大上升子序列长度就是2。而实际上应该是3(子序列[1, 3, 4]),结果不对。
三、代码
3.1 动态规划
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
int lengthOfLIS_DP(vector<int> &nums) { int res = 1; size_t i, j; vector<int> dp(nums.size(), 1); if (nums.empty()) return 0; for (i = 0; i < nums.size(); i++) { // 遍历所有比当前数字小的元素 for (j = 0; j < i; j++) { if (nums[j] < nums[i]) { // 更新dp数组 dp[i] = max(dp[i], dp[j] + 1); // 更新结果 res = max(dp[i], res); } } } return res; } |
3.2 二分法
通过二分法找到区间内第一个大于目标的元素:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
/* * 在区间[left, right]找到第一个大于等于目标的元素 * @v 待查找的数组 * @left 左边界,包含 * @right 右边界,包含 * @return 找到返回元素的下标,否则返回-1 */ int find_first_ge(vector<int> &v, int left, int right, int target) { int mid; while (left <= right) { mid = (left + right) / 2; if (v[mid] >= target) { if (mid == left || v[mid - 1] < target) return mid; right = mid - 1; } else { left = mid + 1; } } return -1; } |
求最长上升子序列:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
int lengthOfLIS_BINSEARCH(vector<int> &nums) { vector<int> tails(nums.size(), 0); int i, idx, max_len; if (nums.empty()) return 0; max_len = 0; for (i = 0; i < nums.size(); i++) { idx = find_first_ge(tails, 0, max_len - 1, nums[i]); if (idx < 0) { // 没有找到一个比当前元素小的,扩充 tails[max_len++] = nums[i]; } else { // 找到了,替换掉 tails[idx] = nums[i]; } } return max_len; } |
评论