703

Author: TedTran2019

ruby/703-Kth-Largest-Element-in-a-Stream.rb

@@ -0,0 +1,16 @@
+require 'rb_heap'
+
+class KthLargest
+  def initialize(k, nums)
+    @k = k
+    @heap = Heap.new
+    nums.each { |num| @heap << num }
+    @heap.pop until @heap.size <= @k
+  end
+
+  def add(val)
+    @heap << val
+    @heap.pop until @heap.size <= @k
+    @heap.peak
+  end
+end