旺崽的博客

要么天赋异禀,要么天道酬勤

0%

LeetCode 354 俄罗斯套娃信封问题

题目链接

给定一些标记了宽度和高度的信封,宽度和高度以整数对形式 (w, h) 出现。当另一个信封的宽度和高度都比这个信封大的时候,这个信封就可以放进另一个信封里,如同俄罗斯套娃一样。

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

说明:

1
不允许旋转信封。

示例:

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

非常经典的二维最长上升子序列,对一维的最长上升子序列,我们很熟悉,但是二维的就必须要排序,那么有下面两种排序方案:

  • 一维正序,二维正序
  • 一维正序,二维逆序

相信绝大部分人都会选择第一种排序方案,但是这样一来算法复杂度就会达到 $O(n^2)$(对每一个信封,必须遍历此信封之前的所有信封才能得到最优解),为了降低时间复杂度,我们可以选择第二种排序方案:
即用一个数组装信封,先装入第一个信封,然后对之后的信封 $i$ 直接二分答案数组,判断里面是否有宽高都大于该信封的信封 $j$,如果没有,证明 $i$ 此时是宽高最大的信封,直接装入答案数组;若有,我们就可以用 $i$ 替换 $j$ 的位置以获得更长的俄罗斯套娃,最后信封数组的长度就是答案,算法复杂度 $O(nlogn)$,AC代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
public:
static bool cmp(vector<int> &a, vector<int> &b) {
if (a[0] != b[0]) return a[0] < b[0];
return a[1] > b[1];
}

int maxEnvelopes(vector<vector<int>> &envelopes) {
if (envelopes.size() == 0) return 0;
sort(envelopes.begin(), envelopes.end(), cmp);
vector<int> temp;
temp.push_back(envelopes[0][1]);
int n = envelopes.size();
for (int i = 1; i < n; i++) {
auto it = lower_bound(temp.begin(), temp.end(), envelopes[i][1]);
if (it == temp.end()) temp.push_back(envelopes[i][1]);
else *it = envelopes[i][1];
}
return temp.size();
}
};