2024-11-07 v. 6.9.8: added "398. Random Pick Index"

README.md

@@ -563,3 +563,4 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/).
 | 382. Linked List Random Node                                    | [Link](https://leetcode.com/problems/linked-list-random-node/)                                    | [Link](./lib/medium/382_linked_list_random_node.rb)                                    | [Link](./test/medium/test_382_linked_list_random_node.rb)                                    |
 | 388. Longest Absolute File Path                                 | [Link](https://leetcode.com/problems/longest-absolute-file-path/)                                 | [Link](./lib/medium/388_longest_absolute_file_path.rb)                                 | [Link](./test/medium/test_388_longest_absolute_file_path.rb)                                 |
 | 394. Decode String                                              | [Link](https://leetcode.com/problems/decode-string/)                                              | [Link](./lib/medium/394_decode_string.rb)                                              | [Link](./test/medium/test_394_decode_string.rb)                                              |
+| 398. Random Pick Index                                          | [Link](https://leetcode.com/problems/random-pick-index/)                                          | [Link](./lib/medium/398_random_pick_index.rb)                                          | [Link](./test/medium/test_398_random_pick_index.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.7'
+  s.version = '6.9.8'
   s.license = 'MIT'
   s.files = ::Dir['lib/**/*.rb'] + %w[README.md]
   s.executable = 'leetcode-ruby'

lib/medium/398_random_pick_index.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+# https://leetcode.com/problems/random-pick-index/
+class RandomPickIndex
+  # @param {Integer[]} nums
+  def initialize(nums)
+    @nums = nums
+    @indices = ::Hash.new { |hash, key| hash[key] = [] }
+  end
+
+  # @param {Integer} target
+  # @return {Integer}
+  def pick(target)
+    if @indices.empty?
+      @nums.each_with_index do |num, index|
+        @indices[num] << index
+      end
+    end
+
+    picks = @indices[target]
+
+    picks[rand(picks.size)]
+  end
+end

test/medium/test_398_random_pick_index.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require_relative '../test_helper'
+require_relative '../../lib/medium/398_random_pick_index'
+require 'minitest/autorun'
+
+class RandomPickIndexTest < ::Minitest::Test
+  def test_default_one
+    random_pick_index = ::RandomPickIndex.new([1, 2, 3, 3, 3])
+    assert([2, 3, 4].include?(random_pick_index.pick(3)))
+    assert([0].include?(random_pick_index.pick(1)))
+    assert([2, 3, 4].include?(random_pick_index.pick(3)))
+  end
+end