406. 根据身高重建队列

Posted 2022-01-01 tu9oh0st

406. 根据身高重建队列

题目描述

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(h, k)表示，其中h是这个人的身高，k是排在这个人前面且身高大于或等于h的人数。 编写一个算法来重建这个队列。

注意：
总人数少于1100人。

示例

输入:
[[7,0], [4,4], [7,1], [5,0], [6,1], [5,2]]

输出:
[[5,0], [7,0], [5,2], [6,1], [4,4], [7,1]]

贴出代码

class Solution 
    public int[][] reconstructQueue(int[][] people) 
        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];
                
            );
            List<int[]> res=new ArrayList<>();
            for(int i=0;i<people.length;i++) 
                res.add(people[i][1],people[i]);
            
            return res.toArray(new int[people.length][]);