算法——根据身高重建队列

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

解题思路:先对数组进行排序,身高为逆序,对于身高相同的元素,大于等于自己的人数为顺序。身高越大的元素优先权越大,之后的元素插进来,对其没有影响。然后进行遍历,因为先被遍历的元素的身高一定是大于后遍历的,然后根据后遍历的人数要求,插入相应的位置即可

class Solution {
    public int[][] reconstructQueue(int[][] people) {
        Arrays.sort(people, (a, b) -> b[0] == a[0] ? a[1] - b[1] : b[0] - a[0]);

        List<int[]> res = new LinkedList<>();

        for(int[] p : people) {
            res.add(p[1], p);
        }

        int n = people.length;
        return res.toArray(new int[n][2]);
    }
}
posted @ 2020-10-31 20:32  lippon  阅读(123)  评论(0编辑  收藏  举报