输入:people = [[7,0],[4,4],[7,1],[5,0],[6,1],[5,2]]
输出:[[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]]
解释:
编号为 0 的人身高为 5 ,没有身高更高或者相同的人排在他前面。
编号为 1 的人身高为 7 ,没有身高更高或者相同的人排在他前面。
编号为 2 的人身高为 5 ,有 2 个身高更高或者相同的人排在他前面,即编号为 0 和 1 的人。
编号为 3 的人身高为 6 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
编号为 4 的人身高为 4 ,有 4 个身高更高或者相同的人排在他前面,即编号为 0、1、2、3 的人。
编号为 5 的人身高为 7 ,有 1 个身高更高或者相同的人排在他前面,即编号为 1 的人。
因此 [[5,0],[7,0],[5,2],[6,1],[4,4],[7,1]] 是重新构造后的队列。
本题描述不太好理解,因此直接贴例子。从例子中可以得知,每个人有两个属性,一个是身高属性,一个是他前面≥他身高的人的人数。给出的数组顺序是不满足people数组中的每个person[1]值的,要重构为满足的形式。跟之前的射气球问题一样,本题我的思路同样是先排序,再贪心。根据身高从高到低排序之后,我原本的思路是遍历原数组,在遍历每个元素时再遍历它前面的元素,统计身高大于该元素的个数并找到位置。
在看到官方题解后得出正确解法:的确应该先按身高从高到低排列,但之后只需新建一个list,然后遍历数组,每个元素根据person[1]的值插入即可,因为是从高到低排序的,因此后插入的值不会影响到先插入的值。
需要注意的点是:不止身高要从高到低排序,相同身高的人要把第二属性小的人排在前面,这样才能保证后插入的元素对先插入的元素没有影响。
class Solution {
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people, new Comparator<int[]>() {
public int compare(int[] o1, int[] o2) {
if (o1[0] != o2[0]) {
return o2[0] - o1[0];
} else {
return o1[1] - o2[1];
}
}
});
List<int[]> ans = new ArrayList<int[]>();
for (int[] person : people) {
ans.add(person[1], person);
}
return ans.toArray(new int[ans.size()][]);
}
}