马戏团人塔

题目描述

来源于 https://leetcode-cn.com/

有个马戏团正在设计叠罗汉的表演节目,一个人要站在另一人的肩膀上。出于实际和美观的考虑,在上面的人要比下面的人矮一点且轻一点。已知马戏团每个人的身高和体重,请编写代码计算叠罗汉最多能叠几个人。

示例:

输入:height = [65,70,56,75,60,68] weight = [100,150,90,190,95,110]
输出:6
解释:从上往下数,叠罗汉最多能叠 6 层:(56,90), (60,95), (65,100), (68,110), (70,150), (75,190)

提示:

  • height.length == weight.length <= 10000

解法:

这里使用的是最长递增子序列的解题思路,关于最长递增子序列,可以参考Longest Increasing Subsequence

class Solution {
public:
    int bestSeqAtIndex(vector<int>& height, vector<int>& weight) {
        vector<vector<int>> nums;
        nums.reserve(height.size());
        for(size_t i=0;i<height.size();i++){
            nums.push_back({height[i], weight[i]});
        }
        sort(nums.begin(), nums.end(), [](auto& a, auto& b){
            if(a[0] < b[0]) return true;
            if(a[0] == b[0]) return a[1] > b[1];
            return false;
        });

        vector<int> dp;
        dp.push_back(nums[0][1]);
        for(size_t i=1;i<nums.size();i++){
            if(nums[i][1] > dp.back()){
                dp.push_back(nums[i][1]);
            }else{
                auto it = lower_bound(dp.begin(), dp.end(), nums[i][1]);
                *it = nums[i][1];
            }
        }
        return dp.size();
    }
};