力扣406题:根据身高重建队列
题目描述
假设有打乱顺序的一群人站成一个队列,数组 people 表示队列中一些人的属性(不一定按顺序)。每个 people[i] = [hi, ki] 表示第 i 个人的身高为 hi ,前面 正好 有 ki 个身高大于或等于 hi 的人。
请你重新构造并返回输入数组 people 所表示的队列。返回的队列应该格式化为数组 queue ,其中 queue[j] = [hj, kj] 是队列中第 j 个人的属性(queue[0] 是排在队列前面的人)。
输入输出样例
输入: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]] 是重新构造后的队列。
输入: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]]
解法一,先排序再插入队列
按照身高从高到低逆序排序,再按照前面的人数从低到高进行排序
vector<vector<int>>reconstructQueue(vector<vector<int>>&people)
{
if(people.empty())
{
return people;
}
//首先先将原始数组先按照身高大小进行降序排序,再按照前面人数个数进行降序排序
int length=people.size();
// 使用冒泡排序的算法
for(int i=0;i<length;i++)
{
for(int j=length-2;j>=i;j--)
{
if(people[j][0]<people[j+1][0])
{
swap(people[j],people[j+1]);
}
else if(people[j][0]==people[j+1][0])
{
if(people[j][1]>people[j+1][1])
{
swap(people[j],people[j+1]);
}
}
}
}
//将排序结果插入到返回的数组中
vector<vector<int>>res;
for(int i=0;i<length;i++)
{
if(res.size()<=people[i][1])
{
res.push_back(people[i]);
}
else{
res.insert(people[i][1]+res.begin(),people[i]);
}
}
return res;
}
void swap(vector<int>&nums1,vector<int>&nums2)
{
vector<int>temp=nums1;
nums1=nums2;
nums2=temp;
}
使用lamda表达式
//使用lamda表达式排序
vector<vector<int>>reconstructQueue2(vector<vector<int>>&people)
{
if(people.empty())
{
return people;
}
//首先先将原始数组先按照身高大小进行降序排序,再按照前面人数个数进行降序排序
int length=people.size();
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>>res;
for(int i=0;i<length;i++)
{
if(res.size()<=people[i][1])
{
res.push_back(people[i]);
}
else{
res.insert(people[i][1]+res.begin(),people[i]);
}
}
return res;
}