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值把他们插入到队列中。
(1)首先找到身高最高的人并对他们进行排序。
(2)然后找到身高次高的人,按照他们的前面的人数把他们插入到最高的人群中。
因此这是一个排序和插入的过程,按照身高进行降序排序,然后把身高相同的人按照k进行升序排序。每次取出身高相同的一组人,按照k值把他们插入到队列中。
public class Solution {
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people, new Comparator<int[]>(){
public int compare(int[] a, int[] b){
if(a[0]==b[0]) return a[1]-b[1];
else return b[0]-a[0];
}
});
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][]);
}
}
几个不熟悉的点
1、Java的Comparator接口实现自定义排序Arrays.sort()
Arrays.sort(nodes, new Comparator<Node>() {
public int compare(Node node1, Node node2) {
if (node1.min == node2.min) {
if (node1.sum == node2.sum)
return node1.index - node2.index;
else
return node2.sum - node1.sum;
}
else
return node2.min - node1.min;
}
});
升序是前者减去后者,降序是后者减去前者。
2、linkedList的add函数
void add(int index, Object element)
Inserts the specified element at the specified position index in this list. Throws IndexOutOfBoundsException if the specified index is out of range (index < 0 || index > size()).
3、二维数组的性质。