文章目錄

• • 1. 題目
  • 2. 解題
  • • 2.1 優先隊列
    • 2.2 雙指針

1. 題目

給定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
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

2.1 優先隊列

• 以時間為軸
• 建立兩個優先隊列，baby 表示未出生的，people表示存活的
• baby 按照出生時間出隊，并 push 進 people，people 按照死亡時間 出隊
• 記錄 people 的最大人數及年份

struct cmp1 {bool operator()(vector<int>& a, vector<int> &b){return a[0] > b[0];//出生時間的小頂堆} }; struct cmp2 {bool operator()(vector<int>& a, vector<int> &b){return a[1] > b[1];//離開時間的小頂堆} }; class Solution { public:int maxAliveYear(vector<int>& birth, vector<int>& death) {priority_queue<vector<int>,vector<vector<int>>,cmp1> baby;priority_queue<vector<int>,vector<vector<int>>,cmp2> people;vector<int> mostP_year(2,0);//人數——年份int time = 2000, maxPeople = 0;for(int i = 0; i < birth.size(); ++i){baby.push({birth[i],death[i]});time = min(time, birth[i]);}while(!baby.empty() || !people.empty()){while(!baby.empty() && baby.top()[0]==time){ //baby出生了，push進peoplepeople.push(baby.top());baby.pop();}if(people.size() > mostP_year[0]){ //更新最大人數mostP_year[0] = people.size();mostP_year[1] = time;}while(!people.empty() && people.top()[1]==time)people.pop();//people離開了，彈出time++;//白駒過隙，惜乎惜乎}return mostP_year[1];} };

• 將 baby 優先隊列改為數組排序，省去 baby 出隊時間

struct cmp2 {bool operator()(vector<int>& a, vector<int> &b){return a[1] > b[1];} }; class Solution { public:int maxAliveYear(vector<int>& birth, vector<int>& death) {vector<vector<int>> baby;priority_queue<vector<int>,vector<vector<int>>,cmp2> people;vector<int> mostP_year(2,0);int time = 2000, maxPeople = 0, i;for(i = 0; i < birth.size(); ++i){baby.push_back({birth[i],death[i]});time = min(time, birth[i]);}sort(baby.begin(),baby.end(),[](vector<int>& a, vector<int> &b){return a[0] < b[0];});i = 0;while(i < baby.size() || !people.empty()){while(i < baby.size() && baby[i][0]==time){people.push(baby[i]);i++;}if(people.size() > mostP_year[0]){mostP_year[0] = people.size();mostP_year[1] = time;}while(!people.empty() && people.top()[1]==time)people.pop();time++;}return mostP_year[1];} };

2.2 雙指針

class Solution { public:int maxAliveYear(vector<int>& birth, vector<int>& death) {sort(birth.begin(), birth.end());//按出生時間排序sort(death.begin(), death.end());//按離開時間排序int i=0, j=0, count = 0, maxAlive = 0, year = 2020;while(i < birth.size()){if(birth[i] <= death[j]){ //新出生+1count++;if(maxAlive < count){maxAlive = count;year = birth[i];}i++;}else{ //離開一人 -1count--;j++;}}return year;} };

總結

以上是生活随笔為你收集整理的程序员面试金典 - 面试题 16.10. 生存人数（自定义优先队列）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。