leetcode406——Queue Reconstruction By Height

題目大意:給出二維數組代表打亂順序的一隊人,每個人由(身高h,排在他前面身高大於等於他的人的個數k)表示,重排隊列。

分析:貪心算法。由於矮子對高個子來說是“看不見”的(矮子插入在高個子前面不會影響高個子的k值),所以採取每次先放高個子的貪心策略,如果身高相同,按k值遞增排序。這樣的貪心策略能確保每一步插入都使得當前隊列是符合要求的。

做法:第一步:對people排序,先按h降序,再按k升序。第二部:逐個插入。將第i個人插入到索引爲k(people[i][1])的位置即可。這樣高個子放完之後,將矮個子直接插入到k的位置,既能滿足矮子前面有k個人,又不會影響高個子之前的插入順序。

代碼:

class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(),people.end(),[](const vector<int>& a,const vector<int>& b){
            if(a[0] == b[0]) return a[1] < b[1];
            else return a[0] > b[0];
        });
        vector<vector<int>> ans;
        for(int i = 0;i < people.size();i++){
            ans.insert(ans.begin() + people[i][1],people[i]);
        }
        return ans;
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章