354. 俄罗斯套娃信封问题
给你一个二维整数数组 envelopes ,其中 envelopes[i] = [wi, hi] ,表示第 i 个信封的宽度和高度。
当另一个信封的宽度和高度都比这个信封大的时候,这个信封就可以放进另一个信封里,如同俄罗斯套娃一样。
请计算 最多能有多少个 信封能组成一组“俄罗斯套娃”信封(即可以把一个信封放到另一个信封里面)。
注意: 不允许旋转信封
示例 1:
输入:envelopes = [[5,4],[6,4],[6,7],[2,3]]
输出:3
解释:最多信封的个数为 3, 组合为: [2,3] => [5,4] => [6,7]。
示例 2:
输入:envelopes = [[1,1],[1,1],[1,1]]
输出:1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/russian-doll-envelopes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析:
首先将信封按宽度从小到大排序,如果宽度相同,则按高度从小到大排序。
之后使用动态规划来做这一道题,
状态转移方程为f[i]=max(f[j]+1,f[i]),即当第j个信封能够放入第i个信封时,总数量f[i]=f[j]+1;而如果f[j]不能放入f[i],则f[i]不变。
之后遍历一遍f数组,把最大的数输出即可。
代码:
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) {
sort(envelopes.begin(),envelopes.end(),cmp);
vector<int>f(5010,1);/*将所有f初始化为1,即不管哪个信封
都至少有一种选法*/
for(int i=1;i<envelopes.size();i++)
for(int j=0;j<i;j++)
if(envelopes[j][0]<envelopes[i][0]&&envelopes[j][1]<envelopes[i][1])
//如果i的宽度和高度都大于j,状态转移
f[i]=max(f[j]+1,f[i]);
int res=0;
for(int i=0;i<envelopes.size();i++)
res=max(res,f[i]);//遍历数组找到最大的方案
return res;
}
};