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]]

思路

  • 先排序,二维数组的一项用vec表示,则vec[0]表示这个人的身高,vec[1]表示这个人前面有几个比他高的。排序的规则是,高个子在前面;个子一样高的时候,前面人少的放到前面。
  • 遍历排好序的数组,当前的vec能保证,前面人的高度肯定大于等于vec[0]。vec[1]表示前面比vec[0]高的人数,因为vec前的人都比他高,所以直接插入到vec[1]数字代表的位置。等于身高相同的vec,因为排序时是按照前面人少的靠前,所以插入后先对位置不变。
  • 在这里插入图片描述

代码

class Solution {
private:
	// 要用static修饰,让cmp属于类,使得cmp不会有默认的this指针参数,使得可以使用在sort里。
    static bool cmp( vector<int>& a, vector<int>& b ) {
        if ( a[0] != b[0] ) return a[0] > b[0];

        return a[1] < b[1];
    }
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        if ( people.size() == 0 ) return people;

        sort( people.begin(), people.end(), cmp );
        
        for ( int i = 1; i < people.size(); ++i ) {
            if ( people[i][1] == i ) continue;

            vector<int> temp = people[i];

            for ( int j = i - 1; j >= temp[1]; --j )
                people[j+1] = people[j];
            
            people[temp[1]] = temp;
        }

        return people;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值