类似题:LeetCode第 452 题:用最少数量的箭引爆气球(C++)_zj-CSDN博客
先写一个简单暴力一点的dp:
class Solution {
public:
int maxEnvelopes(vector<vector<int>>& envelopes) {
if(envelopes.empty()) return 0;
sort(envelopes.begin(), envelopes.end(), [](const vector<int> &a, const vector<int> &b){
return a[0] < b[0] || (a[0] == b[0] && a[1] < b[1]);
});
//dp[i]表示i之前的所有元素中能装进信封i的最大个数
vector<int> dp(envelopes.size(), 1);
for(int i = 1; i < envelopes.size(); ++i){
for(int j = i-1; j >= 0; --j){
if(envelopes[i][1] > envelopes[j][1] && envelopes[i][0] > envelopes[j][0]){
dp[i] = max(1 + dp[j], dp[i]);
}
}
}
return *max_element(dp.begin(), dp.end());
}
};
和之前的题目类似,for循环里面,遍历j的时候,每次都需要全部遍历。
而实际上因为已经排序的关系,可以换种思路,使用二分查找加上维护单调栈的思想。
右元素要降序很重要,不然就很难处理左元素相等的情况。
class Solution {
public:
static bool cmp(const vector<int> &a, const vector<int> &b){
return a[1] < b[1];
}
int maxEnvelopes(vector<vector<int>>& envelopes) {
if(envelopes.empty()) return 0;
//左元素升序,右元素降序
//右元素降序很重要,代表着左元素相同的时候,更大的会放在前面
sort(envelopes.begin(), envelopes.end(), [](const vector<int> &a, const vector<int> &b){
return a[0] < b[0] || (a[0] == b[0] && a[1] > b[1]);
});
vector<vector<int>> dp;
dp.push_back(envelopes[0]);
for(int i = 1; i < envelopes.size(); ++i){
//查找第一个右元素大于当前信封右元素的信封
auto it = lower_bound(dp.begin(), dp.end(), envelopes[i], cmp);
if(it == dp.end()) dp.push_back(envelopes[i]);//没找到
else *it = envelopes[i];
}
return dp.size();
}
};
可以看这位大佬的解释:最长递增子序列扩展到二维而已 - 俄罗斯套娃信封问题 - 力扣(LeetCode)