[LeetCode] 406、根据身高重建队列

题目描述

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示,其中h是这个人的身高,k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。

输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

解题思路

看题解吧看题解。。。

  • 先排序再插入:先排序再插入(我的实现)

    • 排序规则:按照先H高度降序,K个数升序排序

      static bool cmp(const vector<int> &a, const vector<int> &b){
          return a[0] == b[0]? (a[1] < b[1]): (a[0] > b[0]); 
      }
      
    • 遍历排序后的数组,根据K插入到K的位置上

    • 核心思想:高个子先站好位,矮个子插入到K位置上,前面肯定有K个高个子,矮个子再插到前面也满足K的要求

      假设候选队列为 A,已经站好队的队列为 B.

      从 A 里挑身高最高的人 x 出来,插入到 B。 因为 B 中每个人的身高都比 x 要高,因此 x 插入的位置,就是看 x 前面应该有多少人就行了。比如 x 前面有 5 个人,那 x 就插入到队列 B 的第 5 个位置就可以啦。

      // [7,0], [7,1], [6,1], [5,0], [5,2], [4,4]
      // 再一个一个插入。
      // [7,0]
      // [7,0], [7,1]
      // [7,0], [6,1], [7,1]
      // [5,0], [7,0], [6,1], [7,1]
      // [5,0], [7,0], [5,2], [6,1], [7,1]
      // [5,0], [7,0], [5,2], [6,1], [4,4], [7,1]
      

    注:vector支持插入操作:res.insert(res.begin() + e[1], e);

参考代码

class Solution {
public:
    vector<vector<int> > reconstructQueue(vector<vector<int> >& people) {
        int length = people.size();
        if(length == 0)
            return vector<vector<int> >();
        vector<vector<int> > res;
        
        sort(people.begin(), people.end(), cmp);
        
        for(auto e: people){
            res.insert(res.begin() + e[1], e);  // 顺序迭代器也能 +int (but非顺序迭代器不行,如 list的迭代器)
        }
        
        return res;
    }
    
    static bool cmp(const vector<int> &a, const vector<int> &b){
        return a[0] == b[0]? (a[1] < b[1]): (a[0] > b[0]); 
    }
    
};
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值