假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
思路:
- 选出最高的一群人并在子数组(S)中对它们进行排序(相当于对key由大到小排序,如果key相同就按照value有小到大排序)。由于没有其他人群比他们高,因此每个人的下标将与他的key值相同。
- 对于第二高的组(以及其余组),将每一组按照value插入(S)中。
可以这么做的原因在于:由于按照由大到小排序,所以小元素的后插入位置对前面已经排序的大元素无影响。所以可以按照这个方法做。
例如
输入:[[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
步骤1之后的子数列S:[ [7,0] ],[7,1] ]
在第2步之后的子数列S:[[7,0],[6,1],[7,1]]
......
参考代码:
class Solution {
public:
static bool cmp(const pair<int, int>&a, const pair<int, int>&b) {
return a.first > b.first || (a.first == b.first && a.second < b.second);
}
vector<pair<int, int>> reconstructQueue(vector<pair<int, int>>& people) {
vector<pair<int, int>> res;
sort(people.begin(), people.end(), cmp);
for (auto &p : people) {
res.insert(res.begin() + p.second, p);
}
return res;
}
};