题目
假设有打乱顺序的一群人站成一个队列,数组 people
表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [hi, ki]
表示第 i
个人的身高为 hi
,前面 正好 有 ki
个身高大于或等于 hi
的人。
请你重新构造并返回输入数组 people
所表示的队列。返回的队列应该格式化为数组 queue
,其中 queue[j] = [hj, kj]
是队列中第 j
个人的属性(queue[0]
是排在队列前面的人)。
示例 1:
输入:people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出:[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释:
编号为 0 的人身高为 5 ,没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ,没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ,有 2 个身高更高或者相同的人排在他前面,即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
编号为 4 的人身高为 4 ,有 4 个身高更高或者相同的人排在他前面,即编号为 0、1、2、3 的人。
编号为 5 的人身高为 7 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
因此 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。
示例 2:
输入:people = [[6,0],[5,0],[4,0],[3,2],[2,2],[1,4]]
输出:[[4,0],[5,0],[2,2],[3,2],[1,4],[6,0]]
提示:
1 <= people.length <= 2000
0 <= hi <= 10^6
0 <= ki < people.length
-
解题方法
贪心(从大到小)
对数组按高度从大到小排序,高度相同时
ki
从小到大排序。排序后的数组从前向后遍历,将元素按ki
进行插入的时候,后插入的元素不会影响已插入元素的顺序。
此时,局部最优为没插入一个元素都保证当前数组按指定顺序排列。全局最优为插入全部元素后数组按指定规则排列。list
底层为链表实现,在插入元素方面耗时更低。
时间复杂度O(n^2)
,空间复杂度O(n)
。
C++代码:class Solution {
public:
static bool cmp(const vector<int>& a, const vector<int>& b) {
if(a[0]==b[0]) return (a[1]<b[1]);
return a[0]>b[0];
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), cmp);
list<vector<int>>que;
for(int i=0; i<people.size(); i++) {
int pos = people[i][1];
list<vector<int>>::iterator it = que.begin();
while(pos--) it++;
que.insert(it, people[i]);
}
return vector<vector<int>>(que.begin(), que.end());
}
};
贪心(从小到大)
对数组按身高从小到大排列,身高相同的
ki
大优先。此时顺序遍历数组并插入元素的过程中,后面的元素会影响前面的元素顺序,因此插入的过程中,要给后面的元素按照ki
留够足够的空位。
此时,全局最优为没插入一个元素在保证自身顺序的同时都为后续元素留够了足够的空位;全局最优为全部元素插入后数组按照指定规则排列。
时间复杂度O(n^2)
,空间复杂度O(n)
。
C++代码:class Solution {
public:
static bool cmp(const vector<int>& a, const vector<int>& b) {
if(a[0]==b[0]) return (a[1]>b[1]);
return a[0]<b[0];
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(), people.end(), cmp);
vector<vector<int>> que(people.size());
for(int i=0; i<people.size(); i++) {
int pos = people[i][1] + 1;
int idx = 0;
while(pos) {
if(que[idx].empty()) pos--;
idx++;
}
que[--idx] = people[i];
}
return que;
}
};