leetcode 5750. 人口最多的年份

题目 leetcode 5750. 人口最多的年份

给你一个二维整数数组 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 {
public:
    int hash[2051] = {0};
    int maximumPopulation(vector<vector<int>>& logs) {
        int n = logs.size();
        //统计每一年存在的人数
        for(int i=0; i<n; i++){
            for(int j=logs[i][0]; j<logs[i][1]; j++){
                hash[j]++;
            }
        }
        //查找最大人数的最小年份
        int ans = 1950;
        for(int i=1950; i<2050; i++){
            if(hash[i] > hash[ans]){
                ans = i;
            }
        }
        return ans;
    }
};

方案二：差分
可以用数组来记录每个人出生和死亡的年份，如果某一年有人出生，存活数+1；如果某一年有人死亡，存活数-1。

代码

class Solution {
public:
    int hash[101] = {0};
    int maximumPopulation(vector<vector<int>>& logs) {
        int n = logs.size();
        for(auto& p : logs){
            hash[p[0] - 1950]++;
            hash[p[1] - 1950]--;
        }

        int mx = 0;   // 人口数量最大值
        int res = 0;   // 最大值对应的最小下标
        int ans = 0;   // 每一年的人口数量

        for (int i = 0; i < 101; ++i){
            ans += hash[i];
            if (ans > mx){
                mx = ans ;
                res = i;
            }
        }
        return res + 1950;   // 转回对应的年份
    }
};