Merge pull request #2211 from Jay-0331/rust/0703-kth-largest-element-in-a-stream

Create: 0703-kth-largest-element-in-a-stream.rs

Author: tahsintunan

rust/0703-kth-largest-element-in-a-stream.rs

@@ -0,0 +1,31 @@
+use std::{cmp::Reverse, collections::BinaryHeap};
+
+struct KthLargest {
+    min_heap: BinaryHeap<Reverse<i32>>,
+    size: usize,
+}
+
+impl KthLargest {
+    fn new(k: i32, nums: Vec<i32>) -> Self {
+        let mut kth_largest = KthLargest {
+            min_heap: BinaryHeap::new(),
+            size: k as usize,
+        };
+        for n in nums {
+            kth_largest.add(n);
+        }
+        kth_largest
+    }
+
+    fn add(&mut self, val: i32) -> i32 {
+        self.min_heap.push(Reverse(val));
+        if self.min_heap.len() > self.size {
+            self.min_heap.pop();
+        }
+
+        match self.min_heap.peek() {
+            Some(Reverse(min_val)) => *min_val,
+            _ => -1,
+        }
+    }
+}