赞
踩
给定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]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/living-people-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
直接就数组记录人数,最后遍历选择最大的那一年。
class Solution { public: int maxAliveYear(vector<int>& birth, vector<int>& death) { // 获取出生中的最小年份 int min_year = *min_element(birth.begin(), birth.end()); // 获取死亡中的最大年份 int max_year = *max_element(death.begin(), death.end()); // 创建一个年份数组,从最小年份~最大年份 vector<int> human(max_year - min_year + 1, 0); for (int i = 0; i < birth.size(); ++i) { int start = birth[i], end = death[i]; while (start <= end) { ++human[start - min_year]; ++start; } } // 寻找人数最多的年份 int cnt = human[0], offset = 0; for (int i = 0; i < human.size(); ++i) { if (human[i] > cnt) { cnt = human[i]; offset = i; } } return min_year + offset; } };
树状数组:
const int hi = 102; int lowbit(int x) { return x & (-x); } class Solution { vector<int> live; void update(int idx, int delta) { for (; idx < hi; idx += lowbit(idx)) live[idx] += delta; } int query(int idx) { int ans = 0; for (; idx > 0; idx -= lowbit(idx)) ans += live[idx]; return ans; } public: int maxAliveYear(vector<int>& birth, vector<int>& death) { live = vector<int>(hi); for (int i = 0; i < birth.size(); ++i) { update(birth[i] - 1899, 1); update(death[i] - 1898, -1); } int ans = -1, best = 0; for (int i = 1; i <= 101; ++i) { int year = query(i); if (year > best) { best = year; ans = i + 1899; } } return ans; } }; 链接:https://leetcode-cn.com/problems/living-people-lcci/solution/shu-zhuang-shu-zu-by-lucifer1004/
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。