假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/queue-reconstruction-by-height
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
先把比自己矮的安排好再给不比自己矮的留够位置就行。
代码:
class Solution { public: static bool cmp(vector<int> &a,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>> ans(people.size(),vector<int>()); for(int i = 0;i < people.size();i ++) { int k = 0; for(int j = 0;j < ans.size();j ++) { if(ans[j].empty() && k ++ == people[i][1]) { ans[j] = people[i]; break; } } } return ans; } };