leetcode406 - 根据身高重建队列（贪心题都是脑筋急转弯）

介绍

标签：贪心算法

leetcode406 - 根据身高重建队列
难度 中等

406. 根据身高重建队列
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]]

解题思路

思路来自大佬pphdsny的题解

首先将所有人全部从高到矮排序
然后再次遍历将每个人的数据插入列表
插入的位置取决于那个人的k
因为在当前的情况下，所有比他高的人已经入列了
则说明排他前面的所有人已经排好了，他只要往后稍稍
站在k的位置就好了

先排序
[7,0], [7,1], [6,1], [5,0], [5,2], [4,4]
再一个一个插入。
[7,0]
[7,0], [7,1]
[7,0], [6,1], [7,1]
[5,0], [7,0], [6,1], [7,1]
[5,0], [7,0], [5,2], [6,1], [7,1]
[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]

顺便一提，第二次遍历储存用ArrayList比用LinkedList效率更高，可能是数据很小？

代码

class Solution {
    public int[][] reconstructQueue(int[][] people) {

        //首先将每个人按照从高到矮排序
        Arrays.sort(people, (p1,p2)->p1[0]==p2[0]?p1[1] - p2[1] : p2[0] - p1[0]);

        //新建一个list用于插入
        ArrayList <int[]> list = new ArrayList <>();

        //依次遍历people
        for(int person[] : people)
        {
            //在k的位置上插入当前人的身高数据
            list.add(person[1], person);
        }

        return list.toArray(people);
    }
}