LeetCode--生存人数(Python)

it2024-07-12  43

LeetCode–生存人数(Python)

给定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]

题目来源:力扣 链接:https://leetcode-cn.com/problems/living-people-lcci

思路

前一年的生存人数减去前一年死亡人数加上该年出生人数

class Solution: def maxAliveYear(self, birth, death): from collections import defaultdict d = defaultdict(int) for i in range(len(birth)): d[birth[i]] += 1 d[death[i] + 1] -= 1 ret = [0] a = list(d) a.sort() for i in range(len(a)): ret.append(d[a[i]] + ret[-1]) return a[ret.index(max(ret)) - 1]
最新回复(0)