2025-02-14 v. 8.5.7: added "1395. Count Number of Teams"

Author: fartem

README.md

@@ -689,6 +689,7 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/).
 | 1381. Design a Stack With Increment Operation                                       | [Link](https://leetcode.com/problems/design-a-stack-with-increment-operation/)                                       | [Link](./lib/medium/1381_design_a_stack_with_increment_operation.rb)                                       | [Link](./test/medium/test_1381_design_a_stack_with_increment_operation.rb)                                       |
 | 1382. Balance a Binary Search Tree                                                  | [Link](https://leetcode.com/problems/balance-a-binary-search-tree/)                                                  | [Link](./lib/medium/1382_balance_a_binary_search_tree.rb)                                                  | [Link](./test/medium/test_1382_balance_a_binary_search_tree.rb)                                                  |
 | 1387. Sort Integers by The Power Value                                              | [Link](https://leetcode.com/problems/sort-integers-by-the-power-value/)                                              | [Link](./lib/medium/1387_sort_integers_by_the_power_value.rb)                                              | [Link](./test/medium/test_1387_sort_integers_by_the_power_value.rb)                                              |
+| 1395. Count Number of Teams                                                         | [Link](https://leetcode.com/problems/count-number-of-teams/)                                                         | [Link](./lib/medium/1395_count_number_of_teams.rb)                                                         | [Link](./test/medium/test_1395_count_number_of_teams.rb)                                                         |
 | 1400. Construct K Palindrome Strings                                                | [Link](https://leetcode.com/problems/construct-k-palindrome-strings/)                                                | [Link](./lib/medium/1400_construct_k_palindrome_strings.rb)                                                | [Link](./test/medium/test_1400_construct_k_palindrome_strings.rb)                                                |
 | 2116. Check if a Parentheses String Can Be Valid                                    | [Link](https://leetcode.com/problems/check-if-a-parentheses-string-can-be-valid/)                                    | [Link](./lib/medium/2116_check_if_a_parentheses_string_can_be_valid.rb)                                    | [Link](./test/medium/test_2116_check_if_a_parentheses_string_can_be_valid.rb)                                    |
 | 2425. Bitwise XOR of All Pairings                                                   | [Link](https://leetcode.com/problems/bitwise-xor-of-all-pairings/)                                                   | [Link](./lib/medium/2425_bitwise_xor_of_all_pairings.rb)                                                   | [Link](./test/medium/test_2425_bitwise_xor_of_all_pairings.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 = '8.5.6'
+  s.version = '8.5.7'
   s.license = 'MIT'
   s.files = ::Dir['lib/**/*.rb'] + %w[README.md]
   s.executable = 'leetcode-ruby'

lib/medium/1395_count_number_of_teams.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+# https://leetcode.com/problems/count-number-of-teams/
+# @param {Integer[]} rating
+# @return {Integer}
+def num_teams(rating)
+  size = rating.size
+  up = ::Array.new(size, 0)
+  down = ::Array.new(size, 0)
+  result = 0
+
+  (0...size).each do |i|
+    i.downto(0) do |j|
+      if rating[i] > rating[j]
+        up[i] += 1
+        result += up[j]
+      end
+    end
+  end
+
+  (0...size).each do |i|
+    i.downto(0) do |j|
+      if rating[i] < rating[j]
+        down[i] += 1
+        result += down[j]
+      end
+    end
+  end
+
+  result
+end

test/medium/test_1395_count_number_of_teams.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require_relative '../test_helper'
+require_relative '../../lib/medium/1395_count_number_of_teams'
+require 'minitest/autorun'
+
+class CountNumberOfTeamsTest < ::Minitest::Test
+  def test_default_one = assert_equal(3, num_teams([2, 5, 3, 4, 1]))
+
+  def test_default_two = assert_equal(0, num_teams([2, 1, 3]))
+
+  def test_default_three = assert_equal(4, num_teams([1, 2, 3, 4]))
+end