Queue Reconstruction by Height

问题描述:

Suppose you have a random list of people standing in a queue. Each person is described by a pair of integers (h, k), where h is the height of the person and k is the number of people in front of this person who have a height greater than or equal to h. Write an algorithm to reconstruct the queue.

Note:
The number of people is less than 1,100.

Example
- Input:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
- Output:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

解题分析:

给定一组(h,k)的数,第一个数h表示高度,第二个数表示队列中有k个不小于它的高度的数在它前面。

我们考虑使用贪心算法,先对这一组数进行排序,按照h从大到小的顺序,h相同时,k从小到大排序。

由于标准库中的sort函数是从小到大排序的,我们需要自己写一个排序规则。(注意函数一定要带上static属性)

static bool judge(const pair<int, int> a, const pair<int, int> b) {
    return (a.first > b.first || (a.first == b.first && a.second < b.second));
}

排完序后,我们先把高度最大的数插入到vector容器中,这样无论后面的高度怎么插入,都不会影响比它们高度大的数,插入时的位置使用第二个值k。

for (auto person : people) {
    result.insert(result.begin()+person.second, person);
}

源代码:

class Solution {
public:
    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        vector<pair<int, int>> result;
        sort(people.begin(), people.end(), judge);
        for (auto person : people) {
            result.insert(result.begin()+person.second, person);
        }
        return result;
    }
private:
    static bool judge(const pair<int, int> a, const pair<int, int> b) {
        return (a.first > b.first || (a.first == b.first && a.second < b.second));
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值