Merge pull request #761 from fartem/378_Kth_Smallest_Element_in_a_Sorted_Matrix

2024-10-31 v. 6.9.3: added "378. Kth Smallest Element in a Sorted Matrix"

Author: fartem

README.md

@@ -558,3 +558,4 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/).
 | 347. Top K Frequent Elements                                    | [Link](https://leetcode.com/problems/top-k-frequent-elements/)                                    | [Link](./lib/medium/347_top_k_frequent_elements.rb)                                    | [Link](./test/medium/test_347_top_k_frequent_elements.rb)                                    |
 | 371. Sum of Two Integers                                        | [Link](https://leetcode.com/problems/sum-of-two-integers/)                                        | [Link](./lib/medium/371_sum_of_two_integers.rb)                                        | [Link](./test/medium/test_371_sum_of_two_integers.rb)                                        |
 | 377. Combination Sum IV                                         | [Link](https://leetcode.com/problems/combination-sum-iv/)                                         | [Link](./lib/medium/377_combination_sum_iv.rb)                                         | [Link](./test/medium/test_377_combination_sum_iv.rb)                                         |
+| 378. Kth Smallest Element in a Sorted Matrix                    | [Link](https://leetcode.com/problems/kth-smallest-element-in-a-sorted-matrix/)                    | [Link](./lib/medium/378_kth_smallest_element_in_a_sorted_matrix.rb)                    | [Link](./test/medium/test_378_kth_smallest_element_in_a_sorted_matrix.rb)                    |

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 = '6.9.2'
+  s.version = '6.9.3'
   s.license = 'MIT'
   s.files = ::Dir['lib/**/*.rb'] + %w[README.md]
   s.executable = 'leetcode-ruby'

lib/medium/378_kth_smallest_element_in_a_sorted_matrix.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+# https://leetcode.com/problems/kth-smallest-element-in-a-sorted-matrix/
+# @param {Integer[][]} matrix
+# @param {Integer} k
+# @return {Integer}
+def kth_smallest378(matrix, k)
+  n = matrix.length
+  l = matrix[0][0]
+  h = matrix[n - 1][n - 1]
+  m = 0
+  temp = 0
+  count = 0
+
+  while l < h
+    m = l + (h - l) / 2
+    temp = n - 1
+    count = 0
+
+    matrix.each do |mat|
+      temp -= 1 while temp >= 0 && mat[temp] > m
+
+      count += (temp + 1)
+    end
+
+    if count < k
+      l = m + 1
+    else
+      h = m
+    end
+  end
+
+  l
+end

test/medium/test_378_kth_smallest_element_in_a_sorted_matrix.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require_relative '../test_helper'
+require_relative '../../lib/medium/378_kth_smallest_element_in_a_sorted_matrix'
+require 'minitest/autorun'
+
+class KthSmallestElementInASortedMatrixTest < ::Minitest::Test
+  def test_default_one
+    assert_equal(
+      13,
+      kth_smallest378(
+        [
+          [1, 5, 9],
+          [10, 11, 13],
+          [12, 13, 15]
+        ],
+        8
+      )
+    )
+  end
+
+  def test_default_two
+    assert_equal(
+      -5,
+      kth_smallest378(
+        [[-5]],
+        1
+      )
+    )
+  end
+end