354. 俄罗斯套娃信封问题(动态规划)

256 阅读1分钟

//此问题是LIS的变形题,需要对一个维度进行排序后使用LIS
//优化解法是对一个维度进行升序,然后一个维度进行降序,使用二分查找可以优化成O(nlogn)的时间复杂度
class Solution {
    public int maxEnvelopes(int[][] envelopes) {
        if (envelopes.length == 0 || envelopes[0].length == 0) {
            return 0;
        }
        //先对W进行升序排列
       Arrays.sort(envelopes, new Comparator<int[]>() {
           @Override
           public int compare(int[] o1, int[] o2) {
               if (o1[0] == o2[0]) {
                   return o1[1] - o2[1];
               } else {
                   return o1[0] - o2[0];
               }
           }
       });
        int[] dp = new int[envelopes.length];
        int ans = 0;
        //LIS动态规划模板
        for (int i = 0; i < envelopes.length; i++) {
            dp[i] = 1;
            for (int j = 0; j < i; j++) {
                if (envelopes[i][0] > envelopes[j][0] && envelopes[i][1] > envelopes[j][1]) {
                    dp[i] = Math.max(dp[i], dp[j] + 1);
                }
            }
            ans = Math.max(ans, dp[i]);

        }
        return ans;
    }
}