[LeetCode] 406. Queue Reconstruction by Height Java
题目:
Suppose you have a random list of people standing in a queue. Each person is described by a pair of integers (h, k)
, where h
is the height of the person and k
is the number of people in front of this person who have a height greater than or equal to h
. Write an algorithm to reconstruct the queue.
Note:
The number of people is less than 1,100.
Example
Input: [[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]] Output: [[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
题意及分析:给出了一些人的身高和这个人前面身高大于等于他的人数,对这些人进行排队,使得所有人的情况都得到满足。这道题的思路是(贪心算法):
(1)首先找到身高最高的人并对他们进行排序。
(2)然后找到身高次高的人,按照他们的前面的人数把他们插入到最高的人群中。
因此这是一个排序和插入的过程,按照身高进行降序排序,然后把身高相同的人按照k进行升序排序。每次取出身高相同的一组人,按照k值把他们插入到队列中。
代码:
public class Solution { public int[][] reconstructQueue(int[][] people) { Arrays.sort(people, new Comparator<int []>() { //按身高降序排序(h大的在前面),按k的大小升序排列(k小的在前面) public int compare(int[] a, int[] b) { if(a[0] != b[0]) return -a[0]+b[0]; else return a[1]-b[1]; } }); List<int[]> res=new LinkedList<>(); //保存结果 for(int i=0;i<people.length;i++){ int[] peo = people[i]; res.add(peo[1], peo); } return res.toArray(new int[people.length][]); } }
Seen this question in a real interview before