LeetCode 算法:生存人数

245 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第 19 天,点击查看活动详情

生存人数

原题地址

给定 N 个人的出生年份和死亡年份,第 i 个人的出生年份为 birth[i],死亡年份为 death[i],实现一个方法以计算生存人数最多的年份。

你可以假设所有人都出生于 1900 年至 2000 年(含 19002000 )之间。如果一个人在某一年的任意时期处于生存状态,那么他应该被纳入那一年的统计中。例如,生于 1908 年、死于 1909 年的人应当被列入 1908 年和 1909 年的计数。

如果有多个年份生存人数相同且均为最大值,输出其中最小的年份。

示例:

输入:
birth = {1900, 1901, 1950}
death = {1948, 1951, 2000}
输出: 1901

提示:

  • 0 < birth.length == death.length <= 10000
  • birth[i] <= death[i]

思路分析

  1. 使用 people 来表示每一年增长或者减少的人数,以年份为 key,人数为 value 来存储;
  2. 分析题目可以得知,若在出生年份时,该年的人数一定是加一,若在死亡年份时,下一年的人数一定是减一;
  3. 按照 2 的规律,来遍历出生年份和死亡年份,遇到出生年份中的年份,value加一,遇到死亡年份中的年份,在下一年,value减一
  4. 对于对象来说,若 key 为数字,那么在存储的时候会按照数字升序排列,因此我们不需要给 people 按照 key 中的年份再重新排序,直接处理即可;
  5. 定义 total 来存储总人数,max 为较大的人数,res 为人数较大的年份,遍历 peopletotalvalue 值相加,若 total > max,则将当前年份赋值给 res,并且将 total 值赋给 max;
  6. 遍历完成后,将 res 返回即可。

AC 代码

/**
 * @param {number[]} birth
 * @param {number[]} death
 * @resurn {number}
 */
var maxAliveYear = function(birth, death) {
    const people = {}
    for (let i = 0; i < birth.length; i++) {
        people[birth[i]] = (people[birth[i]] || 0) + 1
    }
    for (let i = 0; i < death.length; i++) {
        people[death[i] + 1] = (people[death[i] + 1] || 0) - 1
    }
    let total = 0, max = 0, res
    for (const year of Object.keys(people)) {
        total += people[year]
        if (total > max) {
            res = year
            max = total
        }
    }
    return res
};

结果:

  • 执行结果: 通过
  • 执行用时:112 ms, 在所有 JavaScript 提交中击败了39.56%的用户
  • 内存消耗:43.9 MB, 在所有 JavaScript 提交中击败了86.81%的用户
  • 通过测试用例:70 / 70

END