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]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/queue-reconstruction-by-height
思路:
先对输入的数据进行简单的整理,对所有的整数对进行比较
最终的排序肯定是先给身高高的人排,再给身高低的(k是该h前面的比他高的人的个数嘛,插入小矮个对他不影响)
所以先简单处理一下,对比整数对的左边和右边,左边不同的就降序排,相同的就看右边,按右边升序排
能得到结果 70 71 61 50 52 44
再从前往后,一个一个元素插入到它该去的位置上去(逐个插入到第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]
用冒泡排序
class Solution { public int[][] reconstructQueue(int[][] people) { sort(people); List<int[]> queue = new ArrayList<>(); for (int[] p : people) { queue.add(p[1], p); //参数1表示第k个位置,参数2表示要插入的人(h,k) } return queue.toArray(new int[queue.size()][]); } public void sort(int[][] arr) { //如果左边的不同就降序排列,如果左边的相同就根据右边升序 for (int i = 1; i < arr.length; i++) { for (int j = i; j > 0; j--) { if ((arr[j][0] == arr[j - 1][0] && arr[j][1] < arr[j - 1][1]) || arr[j][0] > arr[j - 1][0]) { int temp = arr[j - 1][0]; arr[j - 1][0] = arr[j][0]; arr[j][0] = temp; int temp1 = arr[j - 1][1]; arr[j - 1][1] = arr[j][1]; arr[j][1] = temp1; } } } } }
用比较器
public int[][] reconstructQueue(int[][] people) { if (people == null || people.length == 0 || people[0].length == 0) { return new int[0][0]; } Arrays.sort(people, (a, b) -> (a[0] == b[0] ? a[1] - b[1] : b[0] - a[0])); List<int[]> queue = new ArrayList<>(); for (int[] p : people) { queue.add(p[1], p); } return queue.toArray(new int[queue.size()][]); }