手撕:最长递增子序列

求解最长递增子序列(Longest Increasing Subsequence, LIS)问题的常见算法有动态规划和二分查找优化的动态规划。下面详细介绍这两种方法。

力扣原题链接

方法一:动态规划

  1. 定义一个数组 dp其中 dp[i] 表示以 nums[i] 结尾的最长递增子序列的长度
  2. 初始化 dp 数组为 1,因为每个元素自身可以构成一个长度为 1 的递增子序列。
  3. 遍历序列,对于每个 nums[i],检查 nums[0]nums[i-1] 中的每个元素 nums[j],如果 nums[j] 小于 nums[i],则更新 dp[i]
  4. 最终答案为 dp 数组中的最大值。
public class Solution {
	/**
     * 求最长递增子序列长度(动态规划)
     */
    public int lengthOfLIS(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        
        int n = nums.length;
        int[] dp = new int[n];
        int maxLength = 1;
        
        for (int i = 0; i < n; i++) {
            dp[i] = 1; // 每个元素自身可以构成一个长度为1的递增子序列
            for (int j = 0; j < i; j++) {
                if (nums[j] < nums[i]) {
                    dp[i] = Math.max(dp[i], dp[j] + 1);
                }
            }
            maxLength = Math.max(maxLength, dp[i]);
        }
        
        return maxLength;
    }
}

时间复杂度为 O ( n 2 ) O(n^2) O(n2),详细解释可以参考:

代码随想录/动态规划/最长递增子序列

方法二:动态规划 + 二分查找

  1. 定义一个数组 dp其中 dp[i] 表示长度为 i+1 的递增子序列的最后一个元素的最小值这里 dp 数组的定义和方法一不同,注意区分)。
  2. 遍历序列,对于每个 nums[i],使用二分查找在 dp 中找到第一个大于等于 nums[i] 的元素,将其替换为 nums[i],如果没有找到,则将 nums[i] 添加到 dp 的末尾。
  3. 最终 dp 的长度即为最长递增子序列的长度。
public class Solution {
	/**
     * 求最长递增子序列长度(动态规划+二分查找)
     */
    public int lengthOfLIS(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        
        int[] dp = new int[nums.length];
        int size = 0;
        
        for (int num : nums) {
            int left = 0, right = size;
            // 二分查找位置
            while (left < right) {
                int mid = left + (right - left) / 2;
                if (dp[mid] < num) {
                    left = mid + 1;
                } else {
                    right = mid;
                }
            }
            // 更新dp数组
            dp[left] = num;
            // 如果num添加到了dp的末尾,增加 size
            if (left == size) {
                size++;
            }
        }
        
        return size;
    }
}

时间复杂度为 O ( n log ⁡ n ) O(n \log n) O(nlogn) ,其中:

  • 二分查找的作用:保持 dp 数组的有序性;提高查找和更新的效率。
  • 更新 dp 数组:如果 num 可以扩展当前最长递增子序列(即 num 大于 dp 的所有元素),则将 num 添加到 dp 的末尾;否则,用 num 替换 dp 中第一个大于或等于 num 的元素,以保证 dp 中每个位置的元素是递增的最小值。

扩展:求最长递增子序列的具体序列结果

只需要记录一下 nums[i] 在长度为 i+1 的递增子序列中的位置,最后逆推便可得到LIS具体序列结果。

class Solution {
    /**
     * 求最长递增子序列的具体结果
     * (动态规划+二分查找)
     */
    public List<Integer> resultOfLIS(int[] nums) {
        int[] dp = new int[nums.length]; // 记录长度为 i+1 的递增子序列的最小末尾元素
        int[] pos = new int[nums.length]; // 记录 nums[i] 在长度为 i+1 的递增子序列中的位置
        int size = 0;

        for (int i = 0; i < nums.length; i++) {
            int left = 0, right = size;
            // 二分查找位置
            while (left < right) {
                int mid = left + (right - left) / 2;
                if (dp[mid] < nums[i]) {
                    left = mid + 1;
                } else {
                    right = mid;
                }
            }
            // 更新 dp 数组
            dp[left] = nums[i];
            // 更新 pos 数组
            pos[i] = left;
            // 如果 num 添加到了 dp 的末尾,增加 size
            if (left == size) {
                size++;
            }
        }

        // 逆推出最长递增子序列
        List<Integer> result = new LinkedList<>();
        for (int i = nums.length - 1; i >= 0; i--) {
            if (pos[i] == size - 1) {
                result.add(0, nums[i]);
                size--;
            }
        }

        return result;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值