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]]
class Solution {
public:
vector<vector<int>> reconstructQueue(vector<vector<int>>& people)
{
//排序:将最高的人按照 k 值升序排序,然后将它们放置到输出队列中与 k 值相等的索引位置上。
//按降序取下一个高度,同样按 k 值对该身高的人升序排序,然后逐个插入到输出队列中
sort(people.begin(), people.end(),[](const vector<int> &a,const vector<int> &b) //lambda表达式
{
if(a[0] > b[0]) return true;//先按身高h排序
if(a[0] == b[0] && a[1] < b[1]) return true;//身高相同按k排序
return false;
});
vector<vector<int>> res;
for( auto &n : people)
{
//每个向量都按其人数k放在对应的位置上(对整体已经排好的向量而言)
res.insert(res.begin()+n[1],n);
}
return res;
}
};