差分数组
面试题 16.10.生存人数
给定 N 个人的出生年份和死亡年份,第 i 个人的出生年份为 birth[i],死亡年份为 death[i],实现一个方法以计算生存人数最多的年份。
你可以假设所有人都出生于 1900 年至 2000 年(含 1900 和 2000 )之间。如果一个人在某一年的任意时期处于生存状态,那么他应该被纳入那一年的统计中。例如,生于 1908 年、死于 1909 年的人应当被列入 1908 年和 1909 年的计数。
如果有多个年份生存人数相同且均为最大值,输出其中最小的年份。
面试题 16.10.生存人数
class Solution:
def maxAliveYear(self, birth: List[int], death: List[int]) -> int:
dp = [0]*102 # 101+1增加一位保证end不越界
for i in range(len(birth)):
start,end = birth[i]-1900, death[i]-1900
dp[start] += 1
dp[end+1] -= 1
t = 0
print(dp)
ma,idx = 0,0
for i in range(101):
t += dp[i]
if t>ma:
ma = t
idx = i
return idx+1900