力扣406. 根据身高重建队列

最新推荐文章于 2024-03-24 20:00:51 发布

原创最新推荐文章于 2024-03-24 20:00:51 发布 · 280 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#队列 #算法 #数据结构 #python #java

本文介绍了一种解决特定队列重建问题的方法，通过二维数组的排序和遍历技巧，实现了根据身高和前排高人数量重建队列的功能。文章详细解析了算法步骤，包括按身高降序和k值升序排序，以及如何正确插入元素到队列中。

下面这题的解法不仅仅让我学会了解这道题，也让我见识到了二维数组竟然还能以一行元素作为单位进行排序和遍历操作，真是学到了，学到了~~~

406. 根据身高重建队列

假设有打乱顺序的一群人站成一个队列。每个人由一个整数对(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]]

题意分析：

其实这题就是说有一个队列，这个队列中高度层次不齐，但是知道每个人的高度 p[0] 以及他前面比他高的人的个数 p[1] ，本来如果这个people按照队列的顺序给我们每个人的这个p[]数组，那输入就是答案。但是现在题目把每个人的p[]数组顺序打乱了，让我们求每个人在队列中的位置，就是要把打乱的每个p[]顺序还原

思路：

1. 先按身高降序排序，相同身高按k升序排序，经过此次排序后高的人一定在矮的人的前面并且相同高度的人的相对顺序就是最终结果的相对顺序。请记住这两点，敲黑板

2. 创建一个集合，这个集合的每个元素是一个一维数组，也就是我们二维数组的一行。

3. 以行为单位遍历排好序的people[][]数组，假设每行数据是p[], 把每行元素插入到集合的索引为p[1]的位置，

4. 把集合中的数据转换为一个二维数组，返回即是正确结果

下面解释为什么经过第三步后就达到了我们题目要求的输出结果：

经过第一步排序后高的人一定在矮的人的前面并且相同高度的人的相对顺序就是最终结果的相对顺序，所以在进行第三步的过程中，高的人的数据肯定是先被存入集合的，所以每当我们取出一行数据，集合中已有的元素的身高肯定都是大于等于当前元素的身高的，所以当我们取出p[]数组后，发现前面应该有p[1]个人比自己高或者高度和自己相同，那么当前元素就应该排在集合的p[1]下标的位置(仔细想想是不是)，好比说目前有一个队列的人，这些人要么比你高，要么和你一样高，现在要你插入入队中，保证你前面有p[1]个人的身高大于等于你，你是不是应该排在索引为p[1]的位置

所以经过第三步的插入操作后，把每个人都插入到了正确的位置，所以根据这个集合转换的二维数组当然就是正确结果了。

 1 class Solution {
 2     public int[][] reconstructQueue(int[][] people) {
 3         // 先按身高降序排序，相同身高按k升序排序
 4         Arrays.sort(people, new Comparator<int[]>(){        // 排序的每个元素是一个一维数组，也就是原二维数组的一行数据
 5             public int compare(int[] o1, int[] o2){         // 二维数组排序还能这么排，真是学到了，学到了
 6                 return o1[0] == o2[0] ? o1[1] - o2[1] : o2[0] - o1[0];
 7             }
 8         });
 9 
10         // 遍历数组，把每行元素添加到一个集合的p[1]位置
11         List<int[]> list = new LinkedList<int[]>();     // 把二维数组一行元素作为一个对象
12         for(int[] p : people){
13             list.add(p[1], p);
14         }
15 
16         int n = list.size();
17         // 返回集合转换为数组的数组
18         return list.toArray(new int[n][2]); // 将集合转换成二维数组
19     }
20 }

力扣测试时间为10ms, 空间为40.9MB