题目大意:给出二维数组代表打乱顺序的一队人,每个人由(身高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;
}
};