【力扣】面試16.10 生存人數

題目描述

給定N個人的出生年份和死亡年份，第i個人的出生年份爲birth[i]，死亡年份爲death[i]，實現一個方法以計算生存人數最多的年份。

你可以假設所有人都出生於1900年至2000年（含1900和2000）之間。如果一個人在某一年的任意時期都處於生存狀態，那麼他們應該被納入那一年的統計中。例如，生於1908年、死於1909年的人應當被列入1908年和1909年的計數。

如果有多個年份生存人數相同且均爲最大值，輸出其中最小的年份。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/living-people-lcci
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

算法思路

FOOL

做5412. 在既定時間做作業的學生人數時想到的……

class Solution:
    def maxAliveYear(self, birth: List[int], death: List[int]) -> int:
        def he(y):
            return sum(1 for i,j in zip(birth,death) if i<=y<=j)
        MAX=0
        year=0
        for i in range(min(birth),max(death)+1):
            k=he(i)
            if k>MAX:
                MAX=k
                year=i
        return year

執行用時 :784 ms, 在所有 Python3 提交中擊敗了12.61%的用戶
內存消耗 :14.3 MB, 在所有 Python3 提交中擊敗了100.00%的用戶

NAIVE

class Solution:
    def maxAliveYear(self, birth: List[int], death: List[int]) -> int:
        dp=[0]*101
        for i,j in zip(birth,death):
            for k in range(i-1900,j-1900+1):
                dp[k]+=1
        return dp.index(max(dp))+1900

執行用時 :312 ms, 在所有 Python3 提交中擊敗了43.70%的用戶
內存消耗 :14.6 MB, 在所有 Python3 提交中擊敗了100.00%的用戶

PERFECT | 前綴和

用兩個數組保存從1900年到2000年間每年的出生人口和死亡人口，最大生存人數就是總的生成人口-總的死亡人口，唯一需要注意的是 應該是第i年的總出生人口-第i-1年的死亡人口，因爲當年死亡的人仍算存活。

class Solution:
    def maxAliveYear(self, birth: List[int], death: List[int]) -> int:
        dic = collections.defaultdict(int)
        
        for c in birth:
            dic[c]+=1
        for c in death:
            dic[c+1]-=1
        
        maxn = 0
        cur =0
        res = -1
        for i in range(1900,2001):
            cur+=dic[i]
            if cur>maxn:
                res=i
                maxn = cur
                
        return res

執行用時 :96 ms, 在所有 Python3 提交中擊敗了67.59%的用戶
內存消耗 :14.5 MB, 在所有 Python3 提交中擊敗了100.00%的用戶