给定一些标记了宽度和高度的信封,宽度和高度以整数对形式 (w, h)
出现。当另一个信封的宽度和高度都比这个信封大的时候,这个信封就可以放进另一个信封里,如同俄罗斯套娃一样。
请计算最多能有多少个信封能组成一组“俄罗斯套娃”信封(即可以把一个信封放到另一个信封里面)。
说明:
- 不允许旋转信封。
示例:
输入: envelopes = [[5,4],[6,4],[6,7],[2,3]]
输出: 3
解释: 最多信封的个数为 3, 组合为: [2,3] => [5,4] => [6,7]。
解答
先按照信封的长度进行排序,然后动态规划,o(n^2)
复杂度:
class Solution {
public:
static bool cmp(const vector<int>& a, const vector<int>& b){
return a[0] < b[0];
}
int maxEnvelopes(vector<vector<int>>& envelopes) {
sort(envelopes.begin(), envelopes.end(), cmp);
int len = envelopes.size();
vector<int> dp(len, 1);
for(int i = 0; i < len; i++){
for(int j = 0; j < i; j++){
if(envelopes[i][0] > envelopes[j][0] && envelopes[i][1] > envelopes[j][1])
dp[i] = max(dp[j] + 1, dp[i]);
}
}
return *max_element(dp.begin(), dp.end());
}
};
参考官方题解,类似于单调栈,结合二分搜索去掉一重循环,复杂度降为o(nlogn)
:
class Solution {
public:
static bool cmp(const vector<int>& a, const vector<int>& b){
// 注意这里如果信封长度相等,则按宽度降序排列
// 如 [6, 3] 和[6, 4],实际上是不满足递增关系的
return (a[0] < b[0]) || (a[0] == b[0] && a[1] > b[1]);
}
int maxEnvelopes(vector<vector<int>>& envelopes) {
sort(envelopes.begin(), envelopes.end(), cmp);
int len = envelopes.size();
// record 记录包含当前信封envelopes[j]所形成的最长递增子序列
vector<int> record{envelopes[0][1]};
for(int j = 1; j < len; j++){
int width = envelopes[j][1];
if(width > record.back()){
record.push_back(width);
}
else{
// 二分查找第一个小于等于width的元素
auto index = lower_bound(record.begin(), record.end(), width);
*index = width;
}
}
return record.size();
}
};