leetcode中等406根据身高重建队列

Posted 2021-12-12 qq_40707462

思路：
身高从高到底排序，身高一样时，人数少的在前面
对于当前该去排队的人，已经排好的人全是比他高的，所以只要看他的 k 是多少，直接插入已排好的队里即可

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        if (0 == people.length || 0 == people[0].length)
            return new int[0][0];
        //身高降序，人数升序
        Arrays.sort(people,(int[]a,int[]b)->{
            if(a[0]-b[0]!=0) return b[0]-a[0];
            else return a[1]-b[1];
        });
        List<int[]> res = new ArrayList<>();
        for(int[] p:people){
            res.add(p[1],p);
        }
        return res.toArray(new int[res.size()][2]);
    }
}