diff options
author | Jack Sangdahl <[email protected]> | 2025-03-10 00:36:13 -0600 |
---|---|---|
committer | Jack Sangdahl <[email protected]> | 2025-03-10 00:36:13 -0600 |
commit | 7c60dee83b436e8e8a667bf9f191beced8de4fc3 (patch) | |
tree | eb7a24ee2067f92245c57bb118e85a1baf15df50 /other/kth_element_in_stream.cpp | |
parent | 5d7cdc215e4ba1ec4cbb24c084cccb5e5e641468 (diff) | |
download | leetcode-7c60dee83b436e8e8a667bf9f191beced8de4fc3.tar.gz leetcode-7c60dee83b436e8e8a667bf9f191beced8de4fc3.zip |
chore: move cpp and rust code to directory
Diffstat (limited to 'other/kth_element_in_stream.cpp')
-rw-r--r-- | other/kth_element_in_stream.cpp | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/other/kth_element_in_stream.cpp b/other/kth_element_in_stream.cpp new file mode 100644 index 0000000..75e5234 --- /dev/null +++ b/other/kth_element_in_stream.cpp @@ -0,0 +1,32 @@ +#include "leetcode.h" + +class KthLargest { +private: + int k; + priority_queue<int, vector<int>, greater<int>> pq; + +public: + KthLargest(int k, vector<int> &nums) { + this->k = k; + auto endPointer = nums.begin() + min(k, nums.size()); + pq = priority_queue<int, vector<int>, greater<int>>(nums.begin(), + endPointer); + for (int i = k; i < nums.size(); ++i) { + pq.push(nums[i]); + pq.pop(); + } + } + int add(int val) { + pq.push(val); + if (pq.size() > k) + pq.pop(); + return pq.top(); + } +}; + +int main() { + vector<int> nums({4, 5, 8, 2}), stream({3, 5, 10, 9, 4}); + KthLargest *obj = new KthLargest(3, nums); + for (auto num : stream) + cout << obj->add(num) << ' '; +} |