面试题 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

提示：

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

Python 解答：
1.直觉的做法

class Solution:
    def maxAliveYear(self, birth: List[int], death: List[int]) -> int:
        lis = [0 for i in range(2000-1900+1)]
        for i in range(len(birth)):
            for j in range(birth[i], death[i]+1):
                lis[j-1900] += 1
        index = 0
        for i in range(len(lis)):
            if lis[i] > lis[index]:
                index = i
        return index+1900

2.前缀和

class Solution:
    def maxAliveYear(self, birth: List[int], death: List[int]) -> int:
        b = [0 for i in range(2000-1900+1)]
        d = [0 for j in range(2000-1900+1)]
        for i in range(len(birth)):
            b[birth[i]-1900] += 1
            d[death[i]-1900] += 1
        for i in range(1, len(b)):
            b[i] += b[i-1]
            d[i] += d[i-1]
        index = 0
        value = b[0]
        for i in range(1, len(b)):
            if b[i]-d[i-1] > value:
                index = i
                value = b[i]-d[i-1]
        return index+1900