力扣:16.10. 生存人数

163 阅读1分钟

「这是我参与11月更文挑战的第21天,活动详情查看:2021最后一次更文挑战

描述

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

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

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

  • 示例 1:
输入:
birth = {1900, 1901, 1950}
death = {1948, 1951, 2000}
输出: 1901
  • 提示:
  • 0 < birth.length == death.length <= 10000
  • birth[i] <= death[i]

解析

只关心出生年份和死亡年份,设置一个记录存活人数的数组,如果某一年出生一个人,对应数组下标的值+1,死亡则-1,求出lives数组的前缀和数组,这个数组中元素的最大值就是最大生存人数。

class Solution {
    public int maxAliveYear(int[] birth, int[] death) {
​
        //使用一个数组记录当前年份的人数增减情况
        int[] nums = new int[201];
​
        //遍历数组,为当前年份所在的坐标+1,为死亡年份的下一年-1;
        for(int i = 0; i < birth.length; i++){
            nums[birth[i] - 1900] += 1;
            nums[death[i] - 1900 + 1] -= 1;
        }
​
        int max = 0;  //记录目前生存人数的最大值
        int sum = 0;  //记录当前年份的人数总和
        int res = birth[0];   //记录结果年份
​
        for(int i = 0; i < nums.length; i++){
            sum += nums[i];   //通过遍历nums数组来记录当前年份人数的总和
            if(sum > max) {
                max = sum;
                res = i;
            }
            //题目要求返回年份更小的那一个,因此需要多做一个判断
            else if(sum == max) res = res > i ? i : res;
        }
        return res + 1900;
    }
}

运行结果:

执行结果:通过

执行用时:1 ms, 在所有 Java 提交中击败了100.00%的用户

内存消耗:39 MB, 在所有 Java 提交中击败了52.36%的用户