介绍
标签:贪心算法
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]]
解题思路
首先将所有人全部从高到矮排序
然后再次遍历将每个人的数据插入列表
插入的位置取决于那个人的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);
}
}