Queue Reconstruction by Height

Leetcode-Algorithm-Greedy-406

题目：

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.
（假定一些人在一个队列里面随机地站着。每一个人都是用一个整数对（h，k）来表示，h表示那个人的身高，k表示在他前面比他高或者身高跟他一样的人的人的数量。设计一个算法重新构造这个队列，使里面的每个人都满足条件。）

注意：

假定表示人的整数对都是合理的。

例子：
输入：[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]].
输出：[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]].

---

题解：
方法：（贪心算法）
想法就是每遍历到一个人，都能把它放到一个合适的位置，即之后每遍历一个人都不会影响前一个人的选择。要做到这一点，必须从身高最大的人开始遍历，然后依次处理身高稍微小一点的。这样就能保证处理的人不会影响到前面的人，就算后面的插的前面去了，也不会违反他要满足的情况。所以算法包括以下两个步骤：
①对人，也就是整数对按照身高由大到小的排序，当身高一样时，第二个元素小的在前面，因为第二个元素大的可能把身高跟他一样的人也计算进去了。
②逐个遍历排好序的队列中的元素，把按照第二个元素的大小把人放到结果队列相应位置。

class Solution {
public:
    vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
        auto comp = [](pair<int, int> &p1, pair<int, int> &p2) {
            return (p1.first > p2.first) ||
                   (p1.first == p2.first && p1.second < p2.second);
        };

        sort(people.begin(), people.end(), comp);
        vector<pair<int, int>> res;
        for (auto &p : people) {
            res.insert(res.begin()+p.second, p);
        }

        return res;
    }
};

分析：
算法第一个步骤是排序，平均时间复杂度为$O(nlogn)$；而第二步只是遍历一次队列，时间复杂度为$O(n)$，因此，总的时间复杂度为$O(nlogn)$。