Description
给你一个二维整数数组 logs ,其中每个 logs[i] = [birthi, deathi] 表示第 i 个人的出生和死亡年份。
年份 x 的 人口 定义为这一年期间活着的人的数目。第 i 个人被计入年份 x 的人口需要满足:x 在闭区间 [birthi, deathi - 1] 内。注意,人不应当计入他们死亡当年的人口中。
返回 人口最多 且 最早 的年份。
示例 1:
输入:logs = [[1993,1999],[2000,2010]]
输出:1993
解释:人口最多为 1 ,而 1993 是人口为 1 的最早年份。
示例 2:
输入:logs = [[1950,1961],[1960,1971],[1970,1981]]
输出:1960
解释:
人口最多为 2 ,分别出现在 1960 和 1970 。
其中最早年份是 1960 。
提示:
1 <= logs.length <= 100
1950 <= birthi < deathi <= 2050
Codes
class Solution(object):
def maximumPopulation(self, logs):
"""
:type logs: List[List[int]]
:rtype: int
"""
a = [0 for i in range(101)]
for i in logs:
a[i[0]-1950] += 1
a[i[1]-1950] -= 1
for i in range(1, 101):
a[i] = a[i] + a[i-1]
print(a)
max1 = 0
max2 = 0
for i in range(101):
if max1 < a[i]:
max1 = a[i]
max2 = i
return max2+1950
思路
差分法
先定义一个数组,初始化数组为一个长度为101(题目中年份的长度)且全为0的数组,遍历每一个人,在每个人的出生年份加1,死亡年份减1
最后对数组a做前缀和操作,最终数组a即为每个年份里存活人的数量。
例如logs = [[1993,1999],[2000,2010]]
a = 0, 0, …,0(从1950开始,故有1993-1950 = 43个0),1,0,0,0,0,0,-1,1,0,0,0,0,0,0,0,0,0,-1,0,…0(2050-2010=40个0)
前缀和操作后:
a = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
最后找出最大值输出即可