假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对 (h, k) 表示,其中 h 是这个人的身高,k 是应该排在这个人前面且身高大于或等于 h 的人数。 例如:[5,2] 表示前面应该有 2 个身高大于等于 5 的人,而 [5,0] 表示前面不应该存在身高大于等于 5 的人。
编写一个算法,根据每个人的身高 h 重建这个队列,使之满足每个整数对 (h, k) 中对人数 k 的要求。
示例:
输入:[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]
输出:[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]
提示:
总人数少于 1100 人。
思路:对数对进行排序,不能两个条件都同时考虑,这样无法做出判断,一般选择先选择其中一个进行排序,然后在对另一个进行插入。
代码:
class Solution {
public int[][] reconstructQueue(int[][] people) {
Arrays.sort(people,(p1,p2) -> p1[0] == p2[0] ? p1[1] - p2[1] : p2[0] - p1[0] );
ArrayList<int[]> res = new ArrayList<int[]>();
for(int[] p : people){
res.add(p[1],p);
}
return res.toArray(new int[res.size()][2]);
}
}
其实这道题让我收获更多的是Lambda表达式,因为用的比较少,所以平时也想不来用法,Lambda表达式是Java8的新特性,允许把函数作为一个方法的参数(函数作为参数传进方法中),本题的Arrays.sort(people,(p1,p2) -> p1[0] == p2[0] ? p1[1] - p2[1] : p2[0] - p1[0] ) sort的第二个参数需要接收某个类对象,该类实现了Comparator接口---由这个对象调用compare方法,执行lambda表达式的代码块。