leetcode 406.根据身高重置队列

leetcode 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) {
        vector<vector<int> > ans(people.size(),vector<int>(2,0x3f3f3f3f));
        sort(people.begin(),people.end(),[&](vector<int> a,vector<int> b){
            return a[0] < b[0];
        });
    
        for(int i = 0;i<people.size();++i){
            int spareCount = -1;
            for(int j = 0;j<people.size();++j){
                if(ans[j][0] >= people[i][0]){
                    spareCount++;
                }
                if(spareCount==people[i][1]){
                    ans[j] = people[i];
                    break;
                }
            }
        }
        return ans;
    } 
};