題目大意:給出二維數組代表打亂順序的一隊人,每個人由(身高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;
}
};