diff --git a/README.md b/README.md index a3c047da..d9450e47 100644 --- a/README.md +++ b/README.md @@ -141,3 +141,4 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/). | 682. Baseball Game | [Link](https://leetcode.com/problems/baseball-game/) | [Link](./lib/easy/682_baseball_game.rb) | | 693. Binary Number with Alternating Bits | [Link](https://leetcode.com/problems/binary-number-with-alternating-bits/) | [Link](./lib/easy/693_binary_number_with_alternating_bits.rb) | | 700. Search in a Binary Search Tree | [Link](https://leetcode.com/problems/search-in-a-binary-search-tree/) | [Link](./lib/easy/700_search_in_a_binary_search_tree.rb) | +| 703. Kth Largest Element in a Stream | [Link](https://leetcode.com/problems/kth-largest-element-in-a-stream/) | [Link](./lib/easy/703_kth_largest_element_in_a_stream.rb) | diff --git a/leetcode-ruby.gemspec b/leetcode-ruby.gemspec index 076b36af..3016bd4c 100644 --- a/leetcode-ruby.gemspec +++ b/leetcode-ruby.gemspec @@ -5,7 +5,7 @@ require 'English' ::Gem::Specification.new do |s| s.required_ruby_version = '>= 3.0' s.name = 'leetcode-ruby' - s.version = '1.3.5' + s.version = '1.3.6' s.license = 'MIT' s.files = ::Dir['lib/**/*.rb'] + %w[bin/leetcode-ruby README.md LICENSE] s.executable = 'leetcode-ruby' diff --git a/lib/easy/703_kth_largest_element_in_a_stream.rb b/lib/easy/703_kth_largest_element_in_a_stream.rb new file mode 100644 index 00000000..727ebfc8 --- /dev/null +++ b/lib/easy/703_kth_largest_element_in_a_stream.rb @@ -0,0 +1,22 @@ +# frozen_string_literal: true + +# https://leetcode.com/problems/kth-largest-element-in-a-stream/ +class KthLargest + # @param {Integer} k + # @param {Integer[]} nums + def initialize(k, nums) + @k = k + @a = nums.sort.reverse.first(k) + end + + # @param {Integer} val + # @@return {Integer} + def add(val) + if @a.size < @k || val > @a.last + @a.insert(@a.bsearch_index { |v| val >= v } || -1, val) + @a.pop if @a.size > @k + end + + @a.last + end +end diff --git a/test/easy/test_703_kth_largest_element_in_a_stream.rb b/test/easy/test_703_kth_largest_element_in_a_stream.rb new file mode 100644 index 00000000..61627311 --- /dev/null +++ b/test/easy/test_703_kth_largest_element_in_a_stream.rb @@ -0,0 +1,17 @@ +# frozen_string_literal: true + +require_relative '../test_helper' +require_relative '../../lib/easy/703_kth_largest_element_in_a_stream' +require 'minitest/autorun' + +class KthLargestElementInAStreamTest < ::Minitest::Test + def test_default + kth_largest = ::KthLargest.new(3, [4, 5, 8, 2]) + + assert_equal(4, kth_largest.add(3)) + assert_equal(5, kth_largest.add(5)) + assert_equal(5, kth_largest.add(10)) + assert_equal(8, kth_largest.add(9)) + assert_equal(8, kth_largest.add(4)) + end +end