Leetcode刷题笔记 354 俄罗斯套娃

俄罗斯信封套娃

题目

给你一个二维整数数组 envelopes ，其中 envelopes[i] = [wi, hi] ，表示第 i 个信封的宽度和高度。

当另一个信封的宽度和高度都比这个信封大的时候，这个信封就可以放进另一个信封里，如同俄罗斯套娃一样。

请计算 最多能有多少个 信封能组成一组“俄罗斯套娃”信封（即可以把一个信封放到另一个信封里面）。

注意：不允许旋转信封。

示例 1：
输入：envelopes = [[5,4],[6,4],[6,7],[2,3]]
输出：3
解释：最多信封的个数为 3, 组合为: [2,3] => [5,4] => [6,7]。

示例 2：
输入：envelopes = [[1,1],[1,1],[1,1]]
输出：1
 

提示：
1 <= envelopes.length <= 105
envelopes[i].length == 2
1 <= wi, hi <= 105

分析

首先，只有外面的信封长宽都大于内部的信封，内部的信封才可以装入外部的信封。因此，需要按照长先进行升序，然后按照宽进行降序。最后，为了处理信封内部的选择问题，可以转换为最长升序序列LIS问题的求解。

方法一

使用动态规划来维护最长升序序列数组，结果超出时间限制。
时间复杂度O($N^2$)
空间复杂度O(N）

class Solution {
    public int maxEnvelopes(int[][] envelopes) {
        //信封数
        int n = envelopes.length;
        if(n == 0){
            return 0;
        }
        //排序,先按照长进行排序，然后按照宽进行排序
        Arrays.sort(envelopes, new Comparator<int[]>(){
            @Override
            public int compare(int[] a, int [] b) {
                return a[0] == b[0] ? b[1] - a[1] : a[0] - b[0];
            }
        });
        //动态规划
        int[] dp = new int[n];
        Arrays.fill(dp, 1);
        int ans = 1;
        for(int i = 1; i < n; ++ i){
            for(int j = 0; j < i; ++ j){
                if(envelopes[j][1] < envelopes[i][1]){
                    dp[i] = Math.max(dp[i], dp[j] + 1);
                }
            }
            ans = Math.max(ans,dp[i]);
        }
        //统计长度
        return ans;
    }
}

进阶

需要对动态规划的遍历加以改进，采用二分查找的方式来用空间来换时间。

class Solution {
    public int maxEnvelopes(int[][] envelopes) {
        int n = envelopes.length;
        if (n == 0) {
            return 0;
        }
        Arrays.sort(envelopes, new Comparator<int[]>() {
            public int compare(int[] e1, int[] e2) {
                if (e1[0] != e2[0]) {
                    return e1[0] - e2[0];
                } else {
                    return e2[1] - e1[1];
                }
            }
        });
        int[] f = new int[n];
        f[0] = envelopes[0][1];
        int p = 0;
        for (int i = 1; i < n; ++i) {
            int num = envelopes[i][1];
            if (num > f[p]) {
                f[++p] = num;
            } else {
                int index = binarySearch(f, p, num);
                f[index] = num;
            }
        }
        return p+1;
    }

    public int binarySearch(int[] f, int p, int target) {
        int l = 0, r = p;
        while (l < r) {
            int mid = (r - l) / 2 + l;
            if (f[mid] < target) {
                l = mid + 1;
            } else {
                r = mid;
            }
        }
        return l;
    }
}