406. 根据身高重建队列 Medium

Leetcode笔记目录

---

一、题目描述

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示，其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。
注意：总人数少于1100人。

示例1：

• 输入: [[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.思想

本题考察的是排序解题，但一次排序并不能得到最终正确序列，需要和插入相结合：

• 显然，身高h相同时，k越小越靠前；k相同时，身高h越小越靠前；
• 一次排序不能满足该条件，可以排序后用类似头插法思想的插入操作：
  • 将数组按身高进行降序排列，若身高相等则按k进行升序排列。
  • 新建一个二维数组，按顺序将原二维数组的每个数组插入，插入位置=k。

2.代码

最初版本(速度慢很多)：

vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(),people.end(),[](vector<int> a,vector<int> b){if(a[0]!=b[0])return a[0]>b[0];else return a[1]<=b[1];});
        vector<vector<int>> result;
        for(int i=0;i<people.size();i++){
            result.insert(result.begin()+people[i][1],people[i]); //错误1
        }
        return result;
    }

• 错误
  • 不了解vector.insert方法；

改进版本(使用list进行插入操作)：

vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        sort(people.begin(),people.end(),[](const vector<int>& a,const vector<int>& b){return a[0]!=b[0]?a[0]>b[0]:a[1]<=b[1];});  //三元运算符+引用传递 速度更快
        list<vector<int>> result;
        for(int i=0;i<people.size();i++){
            auto pos = result.begin();
            advance(pos, people[i][1]); //list不支持随机插入 需要advance找到插入位置
            result.insert(pos,people[i]); 
        }
        return vector<vector<int>>(result.begin(), result.end()); //强制转换
    }

---

三、总结

属于解题思想都不会的题目，本题使用了一些加快编译运行的方法。