刷题力扣面试题 16.10. 生存人数

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/living-people-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

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

题目分析

1. 根据题目描述，计算存活人数最多的年份
2. 先计算每个年份的人数变动，在遍历这些年份，统计经过人数变动后，存活的人数

代码

class Solution {
public:
    int maxAliveYear(vector<int>& birth, vector<int>& death) {
        std::vector<int> change(102, 0);
        for (int i = 0; i < birth.size(); ++i) {
            change[birth[i] - 1900] += 1;
            // 导致数组大小由101变为102
            change[death[i] - 1899] -= 1;
        }
        int year = 0;
        int max = 0;
        int cur = 0;
        // 只统计前101个单位
        for (int i = 0; i < 101; ++i) {
            cur += change[i];
            if (cur > max) {
                year = i;
                max = cur;
            }
        }
        return year + 1900;
    }
};