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_largest_element_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_largest_element_stream.cpp')
-rw-r--r-- | other/kth_largest_element_stream.cpp | 43 |
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 +} |