假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/queue-reconstruction-by-height
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
基本思路:个子矮的相对于个子高的是看不见的,假设你只能看到比你高的脑壳儿。
- 从后往前看,是一个严格递增的序列
- 从前往后看,也是一个严格递增的序列
- 影响K值的只有前面身高大于等于你的
- 同等身高的是按照K从小到大排序的
综上所述,可以先安排身高高的就位,当身高低的插入后不影响之前的结果。具体做法,是按照身高从大到小的顺序排序,同等身高的按照K从小到大排序。然后依次按照k值插入到对应的位置
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(),people.end(),[](const vector<int> &l,const vector<int> &r){
return l[0]==r[0]?l[1]<=r[1]:l[0]>r[0]; //[](){} lambdb表达式
});
list<vector<int>> tmp; //list 插入方便。
for(int i=0;i<people.size();i++){
auto pos=tmp.begin();
advance(pos,people[i][1]); //将pos移动到对应位置
tmp.insert(pos,people[i]);
}
return vector<vector<int>>(tmp.begin(),tmp.end());
}
基本思路:按照k值从小到大排序,k相同时按照身高从小到大排序,每次取k=0,且身高最小的元素放入vector中,然后把剩余的序列中大于该值的k都减去1,再排序,循环上述步骤 。