add: 220408 [cpp]

This commit is contained in:
Lam Haoyin 2022-04-08 08:42:27 +08:00
parent 383c665dce
commit acbb036a56
No known key found for this signature in database
GPG Key ID: 8C089CB1A2B7544F
2 changed files with 22 additions and 1 deletions

21
cpp/2204/220408.cpp Normal file
View File

@ -0,0 +1,21 @@
#include <queue>
#include <vector>
class KthLargest {
private:
int k;
std::priority_queue<int, std::vector<int>, std::greater<>> q;
public:
explicit KthLargest(int k_ = 0, const std::vector<int>& nums = {}) : k(k_) {
q = std::priority_queue<int, std::vector<int>, std::greater<>>(nums.begin(), nums.end());
while (q.size() > k)
q.pop();
}
int add(int val) {
q.push(val);
if (q.size() > k)
q.pop();
return q.top();
}
};

View File

@ -3,4 +3,4 @@ PROJECT(2204)
SET(CMAKE_CXX_STANDARD 23)
ADD_EXECUTABLE(2204 220408-CN.cpp)
ADD_EXECUTABLE(2204 220408.cpp)