欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

406. 根据身高重建队列

程序员文章站 2022-03-07 11:51:54
...

题目

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

示例

输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

思路

  • 先排序,二维数组的一项用vec表示,则vec[0]表示这个人的身高,vec[1]表示这个人前面有几个比他高的。排序的规则是,高个子在前面;个子一样高的时候,前面人少的放到前面。
  • 遍历排好序的数组,当前的vec能保证,前面人的高度肯定大于等于vec[0]。vec[1]表示前面比vec[0]高的人数,因为vec前的人都比他高,所以直接插入到vec[1]数字代表的位置。等于身高相同的vec,因为排序时是按照前面人少的靠前,所以插入后先对位置不变。
  • 406. 根据身高重建队列

代码

class Solution {
private:
	// 要用static修饰,让cmp属于类,使得cmp不会有默认的this指针参数,使得可以使用在sort里。
    static bool cmp( vector<int>& a, vector<int>& b ) {
        if ( a[0] != b[0] ) return a[0] > b[0];

        return a[1] < b[1];
    }
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) {
        if ( people.size() == 0 ) return people;

        sort( people.begin(), people.end(), cmp );
        
        for ( int i = 1; i < people.size(); ++i ) {
            if ( people[i][1] == i ) continue;

            vector<int> temp = people[i];

            for ( int j = i - 1; j >= temp[1]; --j )
                people[j+1] = people[j];
            
            people[temp[1]] = temp;
        }

        return people;
    }
};
相关标签: Leetcode Hot 100