LeetCode No.300 最长上升子序列

366 阅读4分钟

原题链接

给定一个无序的整数数组,找到其中最长上升子序列的长度。

示例:

输入: [10,9,2,5,3,7,101,18]
输出: 4
解释: 最长的上升子序列是 [2,3,7,101],它的长度是 4。

说明:

可能会有多种最长上升子序列的组合,你只需要输出对应的长度即可。 你算法的时间复杂度应该为 O(n²) 。 进阶: 你能将算法的时间复杂度降低到 O(n logn) 吗?

动态规划

定义 dp[i] 为考虑前 i 个元素,以第 i 个数字结尾的最长上升子序列的长度,注意nums[i] 必须被选取。

我们从小到大计算 dp[] 数组的值,在计算 dp[i] 之前,我们已经计算出 dp[0…i−1] 的值,则状态转移方程为:

                dp[i]=max(dp[j])+1,其中0≤j<i且num[j]<num[i]

即考虑往 dp[0…i−1] 中最长的上升子序列后面再加一个nums[i]。由于 dp[j] 代表 nums[0…j] 中以 nums[j] 结尾的最长上升子序列,所以如果能从 dp[j] 这个状态转移过来,那么 nums[i] 必然要大于 nums[j],才能将nums[i] 放在nums[j] 后面以形成更长的上升子序列。

最后,整个数组的最长上升子序列即所有 dp[i] 中的最大值。

                    LIS.length = max(dp[i]),其中 0≤ i < n

题解中有动图

		public int lengthOfLIS(int[] nums) {
        int length = nums.length;
        if (length <= 1) return length;
        
        int[] dp = new int[length];
        dp[0] = 1;
        int maxr = 1;
        for(int i=1;i<length;i++){
            int max = 0;
            for(int j=0;j<i;j++){
                 if (nums[i] > nums[j]) {
                     max = Math.max(max,dp[j]);
                 }
            }
            dp[i] = max+1;
            maxr = Math.max(maxr,dp[i]);
        }
        return maxr;
    }
  • 时间复杂度:O(n²) 其中 nn 为数组 nums 的长度。动态规划的状态数为 n,计算状态 dp[i] 时,需要 O(n) 的时间遍历 dp[0…i−1] 的所有状态,所以总时间复杂度为 O(n²)。
  • 空间复杂度:O(n) 开辟了等长数组存放dp。

贪心+二分查找 O(n logn)解法

考虑一个简单的贪心,如果我们要使上升子序列尽可能的长,则我们需要让序列上升得尽可能慢,因此我们希望每次在上升子序列最后加上的那个数尽可能的小。

基于上面的贪心思路,我们维护一个数组 d[i] ,表示长度为 i 的最长上升子序列的末尾元素的最小值,用 len 记录目前最长上升子序列的长度,起始时 len 为 1,d[1]=nums[0]。

同时我们可以注意到 d[i] 是关于 i 单调递增的。因为如果 d[j]≥d[i] 且 j<i,我们考虑从长度为 i 的最长上升子序列的末尾删除 i−j 个元素,那么这个序列长度变为 j ,且第 j 个元素 x(末尾元素)必然小于 d[i],也就小于 d[j]。那么我们就找到了一个长度为 j 的最长上升子序列,并且末尾元素比 d[j] 小,从而产生了矛盾。因此数组 d[] 的单调性得证。

我们依次遍历数组 nums[] 中的每个元素,并更新数组 d[] 和 len 的值。如果 nums[i]>d[len] 则更新 len=len+1,否则在 d[1…len]中找满足 d[i−1]<nums[j]<d[i] 的下标 i,并更新 d[i]=nums[j]。

根据 dd 数组的单调性,我们可以使用二分查找寻找下标 ii,优化时间复杂度。

最后整个算法流程为:

设当前已求出的最长上升子序列的长度为 len(初始时为 11),从前往后遍历数组 nums,在遍历到 nums[i] 时:

如果 nums[i]>d[len] ,则直接加入到 d 数组末尾,并更新 len=len+1;

否则,在 d 数组中二分查找,找到第一个比 nums[i] 小的数 d[k] ,并更新 d[k+1]=nums[i]。

以输入序列 [0, 8, 4, 12, 2][0,8,4,12,2] 为例:

第一步插入 00,d = [0]d=[0];

第二步插入 88,d = [0, 8]d=[0,8];

第三步插入 44,d = [0, 4]d=[0,4];

第四步插入 1212,d = [0, 4, 12]d=[0,4,12];

第五步插入 22,d = [0, 2, 12]d=[0,2,12]。

最终得到最大递增子序列长度为 33。

		public int lengthOfLIS(int[] nums) {
        int length = nums.length,len=1;
        if (length <= 1) return length;
        
        int[] sort = new int[length+1];
        sort[len] = nums[0];
        
        for(int i=1;i<length;i++){
            if(nums[i] > sort[len]){
                sort[++len] = nums[i];
            }else{
								// 二分查找
                int l = 0, r = len-1, pos = 0;
                while (l <= r) {
                    int mid = (l + r) >> 1;
                    if (nums[i] > sort[mid]) {
                        pos = mid;
                        l = mid + 1;
                    }else {
                        r = mid - 1;
                    }
                }
                sort[pos+1] = nums[i];
            }
        }
        return len;
    }
  • 时间复杂度:O(nlogn)。数组 nums 的长度为 n,我们依次用数组中的元素去更新 d 数组,而更新 d 数组时需要进行 O(logn) 的二分搜索,所以总时间复杂度为 O(nlogn)。
  • 空间复杂度:O(n),需要额外使用长度为 n 的 d 数组。