LeetCode 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.

题意大致为：有一队伍，每个人用一个键值对来表示，键值对的第一个数表示这个人的身高，第二个数表示排在这个人前面的人且身高高于这个人的人数量；题目给定这些键值对，要求返回一个队列，使得该队列中人的排列顺序满足上述性质。

二、解题思路

一开始的思路是：先将人按身高从低到高排，如果身高相同则按第二个数从低到高排，然后每次从第一位开始遍历该数组，一旦找到符合要求的就加入到结果队列中并将其从原数组中删除，这样做是对结果队列的每一位逐一找到正确的人，缺点在于每向结果队列中加进一个人，都需要知道目前队列中的人比他高的有多少，才能决定这个人是不是合适加进去，实现起来比较麻烦。

比较机智而且简洁的算法是：从身高高到低遍历每一个人，当每个人要加进队列的时候，由于比他先进入队列的人都比他高，所以他的键值对的第二个值是多少，他就插入到队列中的第几个，且由于后插入的数都比他小，就算插入位置在他之前，也不会影响到他

三、C++代码

class Solution {
public:
static bool cmp (pair<int, int> a , pair<int, int> b){
if(a.first == b.first )
return a.second < b.second ;
return a.first > b.first ;
}
vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
sort(people.begin(), people.end(), cmp);
vector<pair<int, int>> result;
for(auto j : people)
result.insert(result.begin() + j.second, j) ;
return result;
}
};