Merge pull request #789 from fartem/503_Next_Greater_Element_II

2024-12-03 v. 7.1.6: added "503. Next Greater Element II"

Author: fartem

README.md

@@ -583,3 +583,4 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/).
 | 462. Minimum Moves to Equal Array Elements II                   | [Link](https://leetcode.com/problems/minimum-moves-to-equal-array-elements-ii/)                   | [Link](./lib/medium/462_minimum_moves_to_equal_array_elements_ii.rb)                   | [Link](./test/medium/test_462_minimum_moves_to_equal_array_elements_ii.rb)                   |
 | 468. Validate IP Address                                        | [Link](https://leetcode.com/problems/validate-ip-address/)                                        | [Link](./lib/medium/468_validate_ip_address.rb)                                        | [Link](./test/medium/test_468_validate_ip_address.rb)                                        |
 | 474. Ones and Zeroes                                            | [Link](https://leetcode.com/problems/ones-and-zeroes/)                                            | [Link](./lib/medium/474_ones_and_zeroes.rb)                                            | [Link](./test/medium/test_474_ones_and_zeroes.rb)                                            |
+| 503. Next Greater Element II                                    | [Link](https://leetcode.com/problems/next-greater-element-ii/)                                    | [Link](./lib/medium/503_next_greater_element_ii.rb)                                    | [Link](./test/medium/test_503_next_greater_element_ii.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 = '7.1.7'
+  s.version = '7.1.8'
   s.license = 'MIT'
   s.files = ::Dir['lib/**/*.rb'] + %w[README.md]
   s.executable = 'leetcode-ruby'

lib/medium/503_next_greater_element_ii.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+# https://leetcode.com/problems/next-greater-element-ii/
+# @param {Integer[]} nums
+# @return {Integer[]}
+def next_greater_elements(nums)
+  result = []
+  stack = []
+  (0...(2 * nums.size)).reverse_each do |i|
+    index = i % nums.size
+
+    stack.pop while !stack.empty? && nums[stack.last] <= nums[index]
+
+    result[index] = stack.empty? ? -1 : nums[stack.last]
+    stack << index
+  end
+
+  result
+end

test/medium/test_503_next_greater_element_ii.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require_relative '../test_helper'
+require_relative '../../lib/medium/503_next_greater_element_ii'
+require 'minitest/autorun'
+
+class NextGreaterElementIITest < ::Minitest::Test
+  def test_default_one
+    assert_equal(
+      [2, -1, 2],
+      next_greater_elements(
+        [1, 2, 1]
+      )
+    )
+  end
+
+  def test_default_two
+    assert_equal(
+      [2, 3, 4, -1, 4],
+      next_greater_elements(
+        [1, 2, 3, 4, 3]
+      )
+    )
+  end
+end