题目描述:
假设有打乱顺序的一群人站成一个队列,数组 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 <= 106
0 <= ki < people.length
题目数据确保队列可以被重建
c++代码:
class Solution {
public:
//对二维数组排序,先把vector[i][j]小的排在前面,若vector[i][j]一样大,则把vector[i]小的排在前面
static bool cmp(vector<int> a,vector<int> b){
if(a[1]!=b[1])return a[1]<b[1];
return a[0]>b[0];
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
sort(people.begin(),people.end(),cmp);
int n = people.size();
vector<vector<int>> v;
//插入第一个人
v.emplace_back(people[0]);
if(n==1)return v;
for(int i=1;i<n;i++){
int s = v.size(),sum=0,flag=0;
vector<vector<int>>::iterator it = v.begin();
for(int j=0;j<s;j++){
it=it+1;
if(v[j][0]>=people[i][0])sum=sum+1;
if(sum == people[i][1] && sum!=0){
flag=1;
v.emplace(it,people[i]);
break;
}
}
if(sum > people[i][1])v.emplace(v.begin(),people[i]);
else if(flag==0)v.emplace_back(people[i]);
}
return v;
}
};
此题中的关键信息为:对于每个人,他前面大于等于他高度的人的数量,不会因为他前面插入了新的低个子的人而改变。也就是说,先找到满足高个子人的位置并插入高个子的人,再找位置插入低个子的人即可(因为插入低个子并不会影响高个子,之前高个子的位置仍然满足题目要求)。
具体思路:
第一步:
对people数组people[i] = [hi, ki],ki越小肯定就会排在越前面,所以先按照ki进行从小到大排序。
而对于ki相等的情况,因为插入低个子不影响之前的高个子的ki,但是插入高个子可能会影响低个子的ki,所以再按照hi从大到小排序。
(使用sort()中的cmp()来完成此排序步骤)
第二步:
排序后,依次找出每个人满足”前面 正好 有 ki 个身高大于或等于 hi 的人“的位置,把人插入该位置。
即得到所有人的正确排序。
总结:
(1)二维vector用sort排序(先按照第二维大小排序,再按照第一维大小排序):
bool cmp(vetor<int> a, vector<int> b){
if(a[1]!=b[1])return a[1]<b[1];
return a[0]>b[0];
}
(2)c++11中定义了新的函数,二维或者多维vector,可以直接使用
**emplace_back()**代替push_back()
**emplace_front()**代替push_front()
**emplace()**代替insert()