From 7c60dee83b436e8e8a667bf9f191beced8de4fc3 Mon Sep 17 00:00:00 2001 From: Jack Sangdahl Date: Mon, 10 Mar 2025 00:36:13 -0600 Subject: chore: move cpp and rust code to directory --- other/kth_element_in_stream.cpp | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 other/kth_element_in_stream.cpp (limited to 'other/kth_element_in_stream.cpp') 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, greater> pq; + +public: + KthLargest(int k, vector &nums) { + this->k = k; + auto endPointer = nums.begin() + min(k, nums.size()); + pq = priority_queue, greater>(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 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) << ' '; +} -- cgit v1.2.3