假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
思路:首先按身高从高到矮进行排序。然后利用insert函数,依次进行插入操作。
例如,当前元素为[25,3],已经插入的元素有5个(身高肯定大于等于25),那么,就在索引是3的位置输入[25,3],依次类推
C++
class Solution {
public:
static bool cmp(vector<int>& a, vector<int>& b)
{
if(a[0]==b[0])
{
return a[1]<b[1];
}
else
{
return a[0]>b[0];
}
}
vector<vector<int>> reconstructQueue(vector<vector<int>>& people)
{
int n=people.size();
sort(people.begin(),people.end(),cmp);
vector<vector<int>> res;
for(int i=0;i<n;i++)
{
res.insert(res.begin()+people[i][1],people[i]);
}
return res;
}
};
python
class Solution:
def reconstructQueue(self, people: List[List[int]]) -> List[List[int]]:
n=len(people)
people=sorted(people,key=lambda x:(-x[0],x[1]))
res=[]
for s in people:
res.insert(s[1],s)
return res