406根据身高重建队列

一、前言

标签：贪心算法。

问题来源LeetCode 406 难度：中等。

问题链接：https://leetcode-cn.com/problems/queue-reconstruction-by-height/

 

二、题目

在一个由 0 和 1 组成的二维矩阵内，找到只包含 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]]

 

三、思路

• 排序：

• • 按高度降序排列。
  • 在同一高度的人中，按 k 值的升序排列。

• 逐个地把它们放在输出队列中，索引等于它们的 k 值。
• 返回输出队列

复杂度分析：时间复杂度O(n2)，空间复杂度O(n)

 

四、编码实现

//==========================================================================
/*
* @file    : 406_ReconstructQueue.h
* @label   : 贪心算法
* @blogs   : https://blog.csdn.net/nie2314550441/article/details/107498594
* @author  : niebingyu
* @date    : 2020/07/21
* @title   : 406.根据身高重建队列
* @purpose : 假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
*
* 来源：力扣（LeetCode）
* 难度：中等
* 链接：https://leetcode-cn.com/problems/queue-reconstruction-by-height
*/
//==========================================================================
#pragma once
#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include <assert.h>
using namespace std;

#define NAMESPACE_RECONSTRUCTQUEUE namespace NAME_RECONSTRUCTQUEUE {
#define NAMESPACE_RECONSTRUCTQUEUEEND }
NAMESPACE_RECONSTRUCTQUEUE

class Solution 
{
public:
    vector<vector<int>> reconstructQueue(vector<vector<int>>& people) 
    {
        // 排序
        sort(people.begin(), people.end(),
                [](const vector<int>& lhs, const vector<int>& rhs)
                 {return lhs[0] == rhs[0] ? lhs[1] <= rhs[1] : lhs[0] > rhs[0];});
        int len = people.size();
        list<vector<int>> tmp;

        // 循环插入
        for(int i = 0; i < len; ++i)
        {
            auto pos = tmp.begin();
            advance(pos, people[i][1]);
            tmp.insert(pos, people[i]);
        }

        // 重建vector返回
        return vector<vector<int>>(tmp.begin(), tmp.end());
    }
};

以下为测试代码//
// 测试 用例 START
void test(const char* testName, vector<vector<int>>& people, vector<vector<int>> expect)
{
    Solution s;
    vector<vector<int>> result = s.reconstructQueue(people);

    if (result == expect)
        cout << testName << ", solution passed." << endl;
    else
        cout << testName << ", solution failed. " << endl;
}

// 测试用例
void Test1()
{
    vector<vector<int>> people = { {7,0} ,{4,4},{7,1},{5,0},{6,1},{5,2} };
    vector<vector<int>> expect = { {5,0} ,{7,0},{5,2},{6,1},{4,4},{7,1} };
    test("Test1()", people, expect);
}

NAMESPACE_RECONSTRUCTQUEUEEND
// 测试 用例 END
//
void ReconstructQueue_Test()
{
    cout << "------ start 406.根据身高重建队列 ------" << endl;
    NAME_RECONSTRUCTQUEUE::Test1();
    cout << "------ end 406.根据身高重建队列 --------" << endl;
}

执行结果：