LeetCode 406. Queue Reconstruction by Height

406. 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从大到小来进行排序
• 之所以需要从大到小，是因为在遍历的时候，我么可以把当前元素随便的往前移动。他的前面都是比他大的，这样移动之后不会对前面的元素造成影响。
• 很多初学者疑问为什么数组下标要从0开始那？原因就是这样下标表示的意义就是当前元素前面有多少个元素！Bingo我们正好利用这个性质，k表示的就是他前面有几个比他大的元素。现在他前面的全都是比他大的，那么k是不是就是该元素的下标那。表示他前面有k个比他大的元素。
• 反应到代码上就是先把元素记录下来，并从原数组中删除，再把他插入到指定位置。
• 这里有个小问题，就是对于h相同的元素。比如 (5,2) (5,3) 这一组，题目中认为相等的在掐面的比较大后面的比较小。那么应该先排列谁那？加入我们先排列（5，3）它会被放到下标为3的位置。然后我们排列(5,2)他被放到下标为2的位置。问题来了，当(5,3)前面出现了(5,2)的时候，k1=3 不在正确，因为当(5,2)出现在前面的时候他比后面出现的5要大。所以我们应该先排列k小的，它相对于较大的k在数组的前面，这样我们放置较大k后，不会对他造成影响。这就是为什么Bigger2排序函数中增加了对x.first == y.first的单独处理。

class Solution {public:    static bool Bigger2(const pair<int,int>& x, const pair<int,int>& y){        if(x.first == y.first){//非常重要，如果h相同，必须让k小的在前面，这样先排k小的，之后再排k大的会在k小的后面，就不会造成影响了            return x.second < y.second;        }        return (x.first > y.first);    }    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {        sort(people.begin(), people.end(), Bigger2);        int n = people.size();        for (int i = 0; i < n;i++) {            int expect = people[i].second;            if(expect == i)continue;            pair<int,int> save = people[i];            people.erase(people.begin() + i);//删除原来的元素            people.insert(people.begin()+expect, save);        }        return people;    }};