LeetCode 406. 根据身高重建队列(排序)

1. 题目

假设有打乱顺序的一群人站成一个队列。
每个人由一个整数对(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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 按照身高降序,一样,按k升序
  • 然后按照每个人的k站到第k个位置,就是他前面有k个比他高的
class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
    	sort(people.begin(), people.end(),[&](auto a, auto b){
    		if(a[0] == b[0])
    			return a[1] < b[1];
    		return a[0] > b[0];
    	});//身高降序,一样大就按k升序
    	vector<vector<int>> ans;
    	for(int i = 0; i < people.size(); ++i)
    		ans.insert(ans.begin()+people[i][1], people[i]);
    	return ans;
    }
};

440 ms 24 MB

发布了957 篇原创文章 · 获赞 3192 · 访问量 52万+

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/105749075