Leetcode刷题笔记 406. 根据身高重建队列

时间：2020年11月16日
知识点：线段树优化、贪心、模拟
题目链接：https://leetcode-cn.com/problems/queue-reconstruction-by-height/

题目
假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]

解法

1. 自己模拟一遍就知道怎么排序放人
2. 按照人高低排序 如果一样高 按照前面身高大于等于他的人数排
3. 对于每个人 从左往右找第k个空的位置排进去、
4. 线段树优化

代码

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
bool cmp(vector<int> x,vector<int> y){
    if(x[0] == y[0])
        return x[1] > y[1];
    return x[0] < y[0];
}
class Solution {
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(), people.end(), cmp);
        vector<vector<int>> ans(people.size());
        for(vector<int> x:people){
            int space = 0 ;
            for(int i=0;i<people.size();i++){
                if(ans[i].empty()){
                    space++;
                    if(space == x[1]+1){
                        ans[i] = x;
                        break;
                    }
                }
            }
        }
        return ans;
    }
};
int main()
{
    vector<vector<int>> people;
    vector<int> p1 = {7,0};vector<int> p2 = {4,4};vector<int> p3 = {7,1};
    vector<int> p4 = {5,0};vector<int> p5 = {6,1};vector<int> p6 = {5,2};
    people.push_back(p1);people.push_back(p2);people.push_back(p3);
    people.push_back(p4);people.push_back(p5);people.push_back(p6);
    Solution s;
    vector<vector<int>> ans = s.reconstructQueue(people);
    for(vector<int> peo : ans)
        cout<<peo[0]<<" "<<peo[1]<<endl;
    return  0;
}

今天也是爱zz的一天哦！