<LeetCode>406. 根据身高重建队列

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

注意:
总人数少于1100人。

示例

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

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

解题思路——

1.题目要求进行生成一个新的序列,那我们要看生成的序列的特征是什么。毫无疑问,应该是身高作为优先排序因素,因为身高的大概率在矮的前面,而且一定存在一个数字[max,0],因为最大数前面是没有数字的,所以位置一定是0.

2.当我们把第一个数字按照降序排列之后,我们再来看,对于[7,0]来说,应该插入数组的最前端。对于[7,1]来说,应该插入在下标为1的地方,对于[4,4]来说,应该插入在下标为4的地方。

3.按照1,2的指导思想,我们写出:

        vector<pair<int, int>> res(people.size());
        for(int i = 0;i < people.size();i++)
        {
            res.insert(res.begin()+people[i].second , people[i]);
        }
        res.resize(people.size());
        return res;

但是问题来了,insert函数:比如你数组是空,那么你无法直接把元素插到第2个位置上。也就是说,插入的节点上原先必须有元素(或者说有预留空间)。我们来看看这段代码,一开始插入谁[7,0]还是[7,1]?肯定不可能是[X,i>0],因为如果后面的数字大于0,无法使用insert函数,那一开始插入[7,0]还是[5,0]?当然是[7,0],因为先插入[7.0],因为等下一次按照0插入的时候,[5,0]会自动顶到[7,0]的前面,而这时符合逻辑的,因为7比5大。这里暗含的逻辑就是,当第二个数字相同时,第一个数字应该从大到小排列,这样插入的时候会从下到大排列。

4.按照3的思想,代码如下:

class Solution {
public:
    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        sort(people.begin(),people.end(),[](const pair<int,int> &a,const pair<int,int> &b){
            return a.first > b.first || (a.first == b.first && a.second < b.second);
        });
        vector<pair<int, int>> res(people.size());
        for(int i = 0;i < people.size();i++)
        {
            res.insert(res.begin()+people[i].second , people[i]);
        }
        res.resize(people.size());
        return res;
    }
};
总结:本题最关键的是想到用insert函数,以第二个数字为偏移地址进行插入操作。至于怎么解决插入后的冲突问题,就需要一次排序来解决了。


阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页