LeetCode - 406. 根据身高重建队列

LeetCode - 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]]

解题思路

1. 从高到低排序

1. 将每个人按照身高从大到小进行降序排列，对于身高相同的人以 k 升序排列
2. 按照排完序后的顺序，依次放入队列

因第1步的排序处理，后处理的人均不会影响到当前处理的人 k 值，因此当我们放入第 i 个人时，只需要将其插入队列中，使得他的前面恰好有 k 个人即可。

/**
 * @param {number[][]} people
 * @return {number[][]}
 */
var reconstructQueue = function(people) {
  return people
    	    .sort((a, b) => b[0] - a[0] || a[1] - b[1])
    	    .reduce((pre, cur)=> (pre.splice(cur[1], 0, cur), pre), [])
};

2. 从低到高

1. 按照身高进行升序排列，对于身高相同的人以 k 降序排列
2. 依次进入队列，置于预留了 k 个空位后队列的第一个空位上，即队列当前的第 k + 1 个空位

/**
 * @param {number[][]} people
 * @return {number[][]}
 */
var reconstructQueue = function(people) {
    // 从低到高
    // 按照身高从小到大进行升序排列，对于身高相同的人以 k 降序排列
    people.sort((a, b) => a[0] - b[0] || b[1] - a[1])
    // 总人数
    const n = people.length
    const ans = []
    people.forEach(item => {
        // 此人位于第几个空位 (预留 item[1] 个空位给比其高之人)
        let spaces = item[1] + 1
        for (let i = 0; i < n; i++) {
            // 遇到非空位跳过 (此前入列之人比其低, 不影响 spaces)
            if (ans[i]) continue
            // 此空位可预留
            spaces -= 1
            // 找到此人所在位置, 使其进入队列
            if (!spaces) {
                ans[i] = item
                break
            }
        }
    })
    return ans
};