aboutsummaryrefslogtreecommitdiff
path: root/other/kth_largest_element_stream.cpp
diff options
context:
space:
mode:
authorJack Sangdahl <[email protected]>2025-03-10 00:36:13 -0600
committerJack Sangdahl <[email protected]>2025-03-10 00:36:13 -0600
commit7c60dee83b436e8e8a667bf9f191beced8de4fc3 (patch)
treeeb7a24ee2067f92245c57bb118e85a1baf15df50 /other/kth_largest_element_stream.cpp
parent5d7cdc215e4ba1ec4cbb24c084cccb5e5e641468 (diff)
downloadleetcode-7c60dee83b436e8e8a667bf9f191beced8de4fc3.tar.gz
leetcode-7c60dee83b436e8e8a667bf9f191beced8de4fc3.zip
chore: move cpp and rust code to directory
Diffstat (limited to 'other/kth_largest_element_stream.cpp')
-rw-r--r--other/kth_largest_element_stream.cpp43
1 files changed, 43 insertions, 0 deletions
diff --git a/other/kth_largest_element_stream.cpp b/other/kth_largest_element_stream.cpp
new file mode 100644
index 0000000..3ef8b3c
--- /dev/null
+++ b/other/kth_largest_element_stream.cpp
@@ -0,0 +1,43 @@
+// 703. Kth Largest Element in a Stream
+#include "leetcode.h"
+
+/*
+ * design a class to find the k'th largest element in a stream. note that it is
+ * the k'th largest element in the sorted order, not the k'th distinct element.
+ * implement 'KthLargest' class:
+ * - KthLargest(int k, int[] nums) initialises the objct with the integer 'k'
+ * and the stream of integers 'nums' int add(int val) appends the integer 'val'
+ * to the stream and returns the element representing the k'th largest stream
+ * element
+ */
+
+class KthLargest {
+ priority_queue<int, vector<int>, greater<int>> pq;
+ int size;
+
+public:
+ KthLargest(int k, vector<int> &nums) {
+ size = k;
+ for (int i = 0; i < nums.size(); i++) {
+ pq.push(nums[i]);
+ if (pq.size() > k)
+ pq.pop();
+ }
+ }
+ int add(int val) {
+ pq.push(val);
+ if (pq.size() > size)
+ pq.pop();
+ return pq.top();
+ }
+};
+
+int main() {
+ vector<int> nums = {4, 5, 8, 2};
+ KthLargest *obj = new KthLargest(3, nums);
+ printf("%d\n", obj->add(3)); // expect: 4
+ printf("%d\n", obj->add(5)); // expect: 5
+ printf("%d\n", obj->add(10)); // expect: 5
+ printf("%d\n", obj->add(9)); // expect: 8
+ printf("%d\n", obj->add(4)); // expect: 8
+}