LeetCode 1854. 人口最多的年份

1854. 人口最多的年份

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

---

提示 1

For each year find the number of people whose birth_i ≤ year and death_i > year.

---

提示 2

Find the maximum value between all years.

解法1：差分

Java版：

class Solution {
    public int maximumPopulation(int[][] logs) {
        int[] diff = new int[101];
        int n = logs.length;
        int minv = Integer.MAX_VALUE;
        int maxv = Integer.MIN_VALUE;
        for (int i = 0; i < n; i++) {
            int birth = logs[i][0];
            int death = logs[i][1];
            minv = Math.min(minv, birth);
            maxv = Math.max(maxv, death);
            diff[birth - 1950] += 1;
            if (death <= 2050) {
                diff[death - 1950] -= 1;
            }
        } 
        int maxcount = 0;
        int ans = 0;
        int presum = 0;
        for (int i = minv; i <= maxv; i++) {
            presum += diff[i - 1950];
            if (presum > maxcount) {
                maxcount = presum;
                ans = i;
            }
        }
        return ans;
    }
}

Python3版：

class Solution:
    def maximumPopulation(self, logs: List[List[int]]) -> int:
        diff = [0] * 101
        n = len(logs)
        minv = 2051
        maxv = 1949        
        for birth, death in logs:
            minv = min(minv, birth)
            maxv = max(maxv, death)
            diff[birth - 1950] += 1
            if death <= 2050:
                diff[death - 1950] -= 1
        maxcount = 0
        ans = 0
        presum = 0
        for i in range(minv, maxv + 1):
            presum += diff[i - 1950]
            if presum > maxcount:
                maxcount = presum
                ans = i 
        return ans

复杂度分析

• 时间复杂度：O(m+n)，其中 m 为 logs 的长度，n 为年份的跨度。维护变化量数组 diff 的时间复杂度为 O(m)，遍历维护最大值的时间复杂度为 O(n)。
• 空间复杂度：O(n)，即为变化量数组 diff 的空间开销。