406. 根据身高重建队列
特殊情况下:加入身高h全部一样,手动模拟一边排序过程,然后从一般到特殊。
- 根据身高h降序,位置k升序进行排序
- 将新队列按照k直接插入到结果队列中
//406.
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(),
[](const vector<int>& p1, const vector<int>& p2) {
return p1[0] != p2[0] ? p1[0] >= p2[0] : p1[1] <= p2[1];
});
vector<vector<int>> result;
for (int i = 0; i < people.size(); ++i) {
result.insert(result.begin() + people[i][1], people[i]);
}
return result;
}
class Solution:
def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
people=sorted(people, key=lambda x:(-x[0],x[1]))
res = []
for p in people:
res.insert(p[1],p)
return res