aboutsummaryrefslogtreecommitdiff
path: root/other/kth_element_in_stream.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'other/kth_element_in_stream.cpp')
-rw-r--r--other/kth_element_in_stream.cpp32
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) << ' ';
+}