面试题 16.10. 生存人数

题目

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

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

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

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

分析： 典型差分数组问题，创建一个数组，长度为年份的区间长度，统计对应年份的出生及死亡人数，最后扫描数组，计算出区间内每一年份存活的人数

class Solution:
    def maxAliveYear(self, birth: List[int], death: List[int]) -> int:
        count_list = [0]*102
        for b in birth:
            count_list[b-1900] += 1
        for d in death:
            count_list[d-1900+1] -= 1
        year = 0
        max_num = 0
        cur_num = 0
        for i in range(len(count_list)):
            cur_num += count_list[i]
            if cur_num > max_num:
                year = i + 1900
                max_num = cur_num
        return year