leetcode406. Queue Reconstruction by Height

6次阅读

共计 1218 个字符,预计需要花费 4 分钟才能阅读完成。

题目要求
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]]
假设有一组人站成一堆,每个人都记录下了自己的高度,以及在自己前面有多少个不比自己矮的人。现在请按照这个信息将这组人放在队列中正确的位置上并返回。
思路和代码
刚开始我试图用分治法来解决,因为每一个小队列中,高度最矮且站在自己前面的高个最少的人一定位于 k 位置上。但是这样解决其实复杂化了问题。
可以从另一个角度来想,首先我们知道,同等高度的人,其相对位置一定是根据 k 的值从小到大排列的。即 k 越大,则该同等高度的人一定在另一个同等高度的人后面。如果我们从高到低将人们插入队列中,可以知道,k 的值就该人在当前队列中的下标,如:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
首先将其按照 h 和 k 排序,得出结果:
[[7,0],[7,1],[6,1],[5,1],[5,0],[5,2],[4,4]]

当前结果队列为[]
将 [7,0] 插入下标为 0 的位置上 结果队列[[7,0]]
将 [7,1] 插入下标为 1 的位置上 结果队列[[7,0],[7,1]]
将 [6,1] 插入下标为 1 的位置上 结果队列[[7,0],[6,1],[7,1]]
按照这种规则,依次按照顺序和 k 的值将数据插入结果队列中
代码如下:
public int[][] reconstructQueue(int[][] people) {
int[][] result = new int[people.length][2];
Arrays.sort(people, new Comparator<int[]>() {

@Override
public int compare(int[] o1, int[] o2) {
return o1[0] == o2[0] ? o1[1] – o2[1] : o2[0] – o1[0];
}

});
for(int i = 0 ; i<people.length ; i++) {
int pos = people[i][1];
for (int j = i; j > pos; j–) {
result[j] = result[j – 1];
}
result[pos] = people[i];
}
return result;
}

正文完
 0