题目:
给你一个二维整数数组 envelopes ,其中 envelopes[i] = [wi, hi] ,表示第 i 个信封的宽度和高度。
当另一个信封的宽度和高度都比这个信封大的时候,这个信封就可以放进另一个信封里,如同俄罗斯套娃一样。
请计算 最多能有多少个 信封能组成一组“俄罗斯套娃”信封(即可以把一个信封放到另一个信封里面)。
注意:不允许旋转信封。
例:
输入:envelopes = [[5,4],[6,4],[6,7],[2,3]]
输出:3
解释:最多信封的个数为 3, 组合为: [2,3] => [5,4] => [6,7]。
原题地址:
解题思路:
对二维数组排序,排序的规则是:w由小到大,而w相同时h由大到小。因为w相同的几个信封是不能套在一起的,所以去掉w时保证相同w对应的多个h不会组成子序列。
[[5,4],[6,4],[6,7],[2,3]]排序后:[[2,3],[5,4],[6,7],[6,4]],去掉w,则变为一维数组[3,4,7,4],则题目变成一维数组的最大子序列问题,可参考【动态规划】【leetcode】300. 最长递增子序列
c++代码:
bool cmp(const vector<int>& a, const vector<int>& b) {
if (a[0] < b[0]) return true;
if (a[0] == b[0] && b[1] < a[1]) return true;
return false;
}
class Solution {
public:
int maxEnvelopes(vector<vector<int>>& envelopes) {
int n = envelopes.size();
if (n == 0) return 0;
sort(envelopes.begin(), envelopes.end(), cmp);
vector<int> dp(n, 1);
// 把所有h抽取出来组成一个1维数组,求这个数组的最大子序列
for (int i = 1; i < n; i++) {
for (int k = 0; k < i; k++) {
if (envelopes[k][1] < envelopes[i][1]) {
if (dp[k] + 1 > dp[i]) {
dp[i] = dp[k] + 1;
}
}
}
}
int max = 0;
for (int i = 0; i < n; i++) {
if (dp[i] > max) max = dp[i];
}
return max;
}
};
如果无法理解上面的思路,那么对排序后的二维数组直接dp也可以。
上面代码只需改动2点:
1,cmp函数只需按w由小到大排序,函数内容替换为return a[0] < b[0]; h排序与否没有意义,因为后面的dp是需要全部判断的
2,dp时的if比较条件改为(w,h)同时小于当前,即:envelopes[k][0] < envelopes[i][0] && envelopes[k][1] < envelopes[i][1]。
判断前面也可以增加一个小剪枝:if (envelopes[k][0] >= envelopes[i][0]) break; anyway,运行效率并没有提升。