406. 根据身高重建队列
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
思路:
对于两组数据 person1和 person2,有两种情况:
-
person1[0]=person2[0] ,这种情况下如果 person1[1]比 person2[1]大,那么 person1一定会排在 person2的后面。
-
person1[0] ≠ \neq =person2[0] ,这种情况下则考虑先排 身高高的 这样后面排身高低的时候直接选择合适的位置插入就行。
因此,先对整个数组进行排序,整体按 person[0]从高到低,person[1]从低到高顺序 排序。
然后根据 person[1]再插入,因为 person[1]代表在这个数据之前还有多少个大于或者等于该数据身高的数据,由于我们的排序规则能够使得排在该数据之前的数据的身高都是大于或者等于它的身高的数据。
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), [](const vector<int>& u, const vector<int>& v) {
return u[0] > v[0] || (u[0] == v[0] && u[1] < v[1]);
});
vector<vector<int>> ans;
for (const vector<int>& person: people) {
ans.insert(ans.begin() + person[1], person);
}
return ans;
}