LeetCode 面试题 16.10. 生存人数

一、题目

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

  点击此处跳转题目。

二、C# 题解

  这题目很简单，直接开辟数组记录就好，最后比较人数最大的年份：

public class Solution {
    public int MaxAliveYear(int[] birth, int[] death) {
        int[] data = new int[101];
        for (int i = 0; i < birth.Length; i++)
        for (int j = birth[i]; j <= death[i]; j++)
            data[j - 1900]++;
        
        int ans = 1900, max = 0;
        for (int i = 0; i < 101; i++) {
            if (data[i] <= max) continue;
            max = data[i];
            ans = 1900 + i;
        }
        return ans;
    }
}

• 时间：100 ms，击败 100.00% 使用 C# 的用户
• 内存：48.13 MB，击败 100.00% 使用 C# 的用户

---

  奈何最开始写这道题没看清题目要求，以为 0 < birth <= death <= 10000，于是：

• 重新整理了数据格式，记为 (int birth, int death)。
• 手搓顺序插入数据的函数，比较优先级为 birth > death。
• 按年份顺序查找，每次统计当年人数，最后进行比较。

  一看运行结果，hhh 乐了。

public class Q_16_10
{
    public int MaxAliveYear(int[] birth, int[] death) {
        List<(int birth, int death)> data = new List<(int birth, int death)>();

        int left = birth[0], right = death[0], ans = left, max = 1, begin = 0;
        for (int i = 0; i < birth.Length; i++) {
            Insert(data, (birth[i], death[i]));
            if (birth[i] < left) left = birth[i];
            if (death[i] > right) right = death[i];
        }
        for (var i = 0; i < data.Count; i++) {
            Console.WriteLine($"[{data[i].birth}, {data[i].death}]");
        }
        for (int year = left; year < right; year++) {
            int cnt = 0;
            for (int i = begin; i < data.Count; i++) {
                if (Contain(year, data[i])) cnt++;
                else if (year < data[i].birth) break;
            }
            while (begin < data.Count && year > data[begin].death) begin++;
            if (cnt <= max) continue;
            max = cnt;
            ans = year;
        }
        return ans;
    }

    public void Insert(List<(int birth, int death)> data, (int birth, int death) p) {
        int left = 0, right = data.Count, mid = (left + right) / 2;
        while (left < right) {
            if (data[mid].birth == p.birth) break;
            if (data[mid].birth > p.birth) right = mid;
            else left = mid + 1;
            mid = (left + right) / 2;
        }
        if (mid == data.Count) {
            data.Insert(mid, p);
            return;
        }
        while (data[mid].birth == p.birth) {
            if (mid == 0 || mid == data.Count - 1) break;
            if (p.death > data[mid].death) mid++;
            else if (p.death == data[mid - 1].death && p.death < data[mid].death) mid--;
            else break;
        }
        data.Insert(mid, p);
    }

	public bool Contain(int year, (int birth, int death) p) {
        return p.birth <= year && year <= p.death;
    }
}

• 时间：144 ms，击败 20.00% 使用 C# 的用户
• 内存：50.19 MB，击败 10.00% 使用 C# 的用户