LeetCode:根据身高重建队列

题目如下:

假设有打乱顺序的一群人站成一个队列。 每个人由一个整数对(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]]
思路
  1. 首先对数组排序,根据身高从高到低,同等身高下,根据排在前面的人的人数从少到多排序。

  2. 新建List存储结果,当List数量大于等于排在前面的人的人数时直接添加,否则添加到所对应的index

AC解法
class Solution {
    public int[][] reconstructQueue(int[][] people) {
        if (null == people || people.length == 0 || null == people[0] || people[0].length == 0) {
            return people;
        }
        List<int[]> list = new ArrayList<>();
        Collections.addAll(list, people);
        list.sort(new Comparator<int[]>() {
            @Override
            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[]> res = new ArrayList<>();
        for (int[] ints : list) {
            int height = ints[0];
            int target = ints[1];
            if (target >= res.size()) {
                res.add(ints);
                continue;
            }
            res.add(target, ints);
        }
        int[][] resArray = new int[res.size()][2];
        for (int i = 0; i < res.size(); i ++) {
            resArray[i] = res.get(i);
        }
        return resArray;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值