leetcode_1854 人口最多的年份

121 阅读1分钟

要求

给你一个二维整数数组 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

核心代码

class Solution:
    def maximumPopulation(self, logs: List[List[int]]) -> int:
        logs.sort(key = lambda x: x[0])
        f = [0 for _ in range(2060)]
        for b, d in logs:
            f[b] += 1
            f[d] -= 1
        for i in range(1950, 2051):
            f[i] += f[i-1]
        res_num = 0
        res_id = logs[0][0]
        for i in range(1950, 2051):
            if f[i] > res_num:
                res_num = f[i]
                res_id = i
        return res_id

image.png

解题思路:差分,我们将每一年的数据都记录下来,然后从1950~2051之间进行人数累加,循环找出来人数最多的年即可。