这道题主要涉及动态规划,优化时可以考虑贪心算法和二分查找。
原题
给定一个无序的整数数组,找到其中最长上升子序列的长度。
示例:
输入: [10,9,2,5,3,7,101,18]
输出: 4
解释: 最长的上升子序列是 [2,3,7,101],它的长度是 4。
说明:
- 可能会有多种最长上升子序列的组合,你只需要输出对应的长度即可。
- 你算法的时间复杂度应该为 O(n2) 。
进阶: 你能将算法的时间复杂度降低到 O(n log n) 吗?
解题
暴力法
这也是最基础的想法,利用递归,从每一个数开始,一个一个寻找,只要比选中的标准大,那么就以新的数为起点,继续找。全部找完后,找出最长的序列即可。
也看一下代码:
class Solution {
public int lengthOfLIS(int[] nums) {
// 递归查询
return recursiveSearch(nums, Integer.MIN_VALUE, 0);
}
public int recursiveSearch(int[] nums, int standard, int index) {
if (nums.length == index) {
return 0;
}
// 如果包含当前index的数字,其递增长度
int tokenLength = 0;
if (nums[index] > standard) {
tokenLength = 1 + recursiveSearch(nums, nums[index], index + 1);
}
// 如果不包含当前index的数字,其递增长度
int notTokenLength = recursiveSearch(nums, standard, index + 1);
// 返回较大的那个值
return tokenLength > notTokenLength ? tokenLength : notTokenLength;
}
}
提交之后报超出时间限制
,这个也是预料到的,那么我们优化一下。
记录中间结果
仔细分析一下上面的暴力解法,假设 nums 是:[10,9,2,5,3,7,101,18]
,那么从 7 到 101 这个查找,在2、5、3的时候,都曾经查找过一遍。
那么针对这种重复查找的情况,我们可以用一个二维数组,记录一下中间结果,这样就可以达到优化的效果。比如用int[][] result
标记为记录中间结果的数组,那么result[i][j]
就代表着从 nums[i - 1] 开始,无论包含还是不包含 nums[j] 的最大递增序列长度。这样就能保证不再出现重复计算的情况了。
让我们看看代码:
class Solution {
public int lengthOfLIS(int[] nums) {
// 记录已经计算过的结果
int result[][] = new int[nums.length + 1][nums.length];
for (int i = 0; i < nums.length + 1; i++) {
for (int j = 0; j < nums.length; j++) {
result[i][j] = -1;
}
}
// 递归查询
return recursiveSearch(nums, -1, 0, result);
}
public int recursiveSearch(int[] nums, int preIndex, int index, int[][] result) {
if (nums.length == index) {
return 0;
}
// 如果已经赋值,说明计算过,因此直接返回
if (result[preIndex + 1][index] > -1) {
return result[preIndex + 1][index];
}
// 如果包含当前index的数字,其递增序列最大长度
int tokenLength = 0;
if (preIndex < 0 || nums[index] > nums[preIndex]) {
tokenLength = 1 + recursiveSearch(nums, index, index + 1, result);
}
// 如果不包含当前index的数字,其递增序列最大长度
int notTokenLength = recursiveSearch(nums, preIndex, index + 1, result);
// 返回较大的那个值
result[preIndex + 1][index] = tokenLength > notTokenLength ? tokenLength : notTokenLength;
return result[preIndex + 1][index];
}
}
提交OK,但是结果感人,几乎是最慢的了,无论时间还是空间上,都只打败了`5%`左右的用户,那就继续优化。
### 动态规划
假设我知道了从 nums[0] 到 nums[i] 的最大递增序列长度,那么针对 nums[i + 1],我只要去跟前面的所有数比较一下,找出前面所有数中比 nums[i + 1] 小的数字中最大的递增子序列,再加1就是 nums[i + 1] 对应的最大递增子序列。
这样我只要再记录一个最大值,就可以求出整个数组的最大递增序列了。
让我们看看代码:
class Solution {
public int lengthOfLIS(int[] nums) {
if (nums.length == 0) {
return 0;
}
// 动态规划,之前几个数字中,有几个比当前数小的,不断更新
// 存储中间结果
int[] dp = new int[nums.length];
// 最大值,因为数组中至少有一个,所以最小是1
int max = 1;
// 遍历
for (int i = 0; i < dp.length; i++) {
// 当前下标i的最大递增序列长度
int currentMax = 0;
for (int j = 0; j < i; j++) {
// 如果nums[i]比nums[j]大,那么nums[i]可以加在nums[j]后面,继续构成一个递增序列
if (nums[i] > nums[j]) {
currentMax = Math.max(currentMax, dp[j]);
}
}
// 加上当前的数
dp[i] = currentMax + 1;
max = Math.max(dp[i], max);
}
return max;
}
}
提交OK,执行用时:9 ms
,只战胜了75.15%
的 java 提交,看来还是可以继续优化的。
贪心算法 + 二分查找
贪心算法意味着不需要是最完美的结果,只要针对当前是有效的,就可以了。
我们之前在构造递增序列的时候,其实是在不断根据之前的值进行更新的,并且十分准确。但其实并不需要如此,只要保证序列中每个数都相对较小,就可以得出最终的最大长度。
还是以[10,9,2,5,3,7,101,18,4,8,6,12]
举例:
- 从10到2,都是无法构成的,因为每一个都比之前的小。
- 当以最小的2作为起点后,
2,5
、2,3
都是可以作为递增序列,但明显感觉2,3
更合适,因为3更小。 - 因为7大于3,因此递增序列增长为
2,3,7
。 - 因为101也大于7,因此递增序列增长为
2,3,7,101
。 - 因为18小于101,但是大于7,因此我们可以用18替换101,因为18更小,序列更新为
2,3,7,18
- 此时遇到4,4大于3但是小于7,我们可以用它替换7,虽然此时新的序列
2,3,4,18
并不是真正的结果,但首先长度上没有问题,其次如果出现新的可以排在最后的数,一定是大于4的,因为要先大于现在的最大值18。序列更新为2,3,4,18
。 - 同理,8大于4小于18,替换18,此时新的序列
2,3,4,8
,这样是不是大家开始懂得了这个规律。 - 遇到6之后,更新为
2,3,4,6
。 - 遇到12后,更新为
2,3,4,6,12
。
这样也就求出了最终的结果。
结合一下题目说明
里提到的O(nlogn)
,那么就可以想到二分查找,运用到这里也就是找到当前数合适的位置。
接下来让我们看看代码:
class Solution {
public int lengthOfLIS(int[] nums) {
if (nums.length == 0) {
return 0;
}
// 贪心 + 二分查找
// 一个空数组,用来存储最长递增序列
int[] result = new int[nums.length];
result[0] = nums[0];
// 空数组的长度
int resultLength = 1;
// 遍历
for (int i = 1; i < nums.length; i++) {
int num = nums[i];
// 如果num比当前最大数大,则直接加在末尾
if (num > result[resultLength - 1]) {
result[resultLength] = num;
resultLength++;
continue;
}
// 如果和最大数相等,直接跳过
if (num == result[resultLength - 1]) {
continue;
}
// num比最大值小,则找出其应该存在的位置
int shouldIndex = Arrays.binarySearch(result, 0, resultLength, num);
if (shouldIndex < 0) {
shouldIndex = -(shouldIndex + 1);
}
// 更新,此时虽然得出的result不一定是真正最后的结果,但首先其resultLength不会变,之后就算resultLength变大,也是相对正确的结果
// 这里的更新,只是为了让result数组中每个位置上的数,是一个相对小的数字
result[shouldIndex] = num;
}
return resultLength;
}
}
提交OK,执行用时:2 ms
,差不多了。
总结
以上就是这道题目我的解答过程了,不知道大家是否理解了。这道题目用动态规划其实就已经能解决了,但为了优化,还需要用到贪心算法和二分查找。
有兴趣的话可以访问我的博客或者关注我的公众号、头条号,说不定会有意外的惊喜。
公众号:健程之道
共同学习,写下你的评论
评论加载中...
作者其他优质文章