diff --git a/README.md b/README.md index 6444e9be..e9a4d889 100644 --- a/README.md +++ b/README.md @@ -651,6 +651,7 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/). | 973. K Closest Points to Origin | [Link](https://leetcode.com/problems/k-closest-points-to-origin/) | [Link](./lib/medium/973_k_closest_points_to_origin.rb) | [Link](./test/medium/test_973_k_closest_points_to_origin.rb) | | 979. Distribute Coins in Binary Tree | [Link](https://leetcode.com/problems/distribute-coins-in-binary-tree/) | [Link](./lib/medium/979_distribute_coins_in_binary_tree.rb) | [Link](./test/medium/test_979_distribute_coins_in_binary_tree.rb) | | 981. Time Based Key-Value Store | [Link](https://leetcode.com/problems/time-based-key-value-store/) | [Link](./lib/medium/981_time_based_key_value_store.rb) | [Link](./test/medium/test_981_time_based_key_value_store.rb) | +| 985. Sum of Even Numbers After Queries | [Link](https://leetcode.com/problems/sum-of-even-numbers-after-queries/) | [Link](./lib/medium/985_sum_of_even_numbers_after_queries.rb) | [Link](./test/medium/test_985_sum_of_even_numbers_after_queries.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) | diff --git a/leetcode-ruby.gemspec b/leetcode-ruby.gemspec index 89626899..b2e7496f 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 = '8.1.8' + s.version = '8.1.9' s.license = 'MIT' s.files = ::Dir['lib/**/*.rb'] + %w[README.md] s.executable = 'leetcode-ruby' diff --git a/lib/medium/985_sum_of_even_numbers_after_queries.rb b/lib/medium/985_sum_of_even_numbers_after_queries.rb new file mode 100644 index 00000000..0dee7e71 --- /dev/null +++ b/lib/medium/985_sum_of_even_numbers_after_queries.rb @@ -0,0 +1,28 @@ +# frozen_string_literal: true + +# https://leetcode.com/problems/sum-of-even-numbers-after-queries/ +# @param {Integer[]} nums +# @param {Integer[][]} queries +# @return {Integer[]} +def sum_even_after_queries(nums, queries) + result = ::Array.new(nums.length, 0) + pointer = 0 + sum = nums.select(&:even?).sum + + queries.each do |query| + index = query[1] + num = nums[index] + + sum -= num if num.even? + + nums[index] += query[0] + num = nums[index] + + sum += num if num.even? + + result[pointer] = sum + pointer += 1 + end + + result +end diff --git a/test/medium/test_985_sum_of_even_numbers_after_queries.rb b/test/medium/test_985_sum_of_even_numbers_after_queries.rb new file mode 100644 index 00000000..29ab5d02 --- /dev/null +++ b/test/medium/test_985_sum_of_even_numbers_after_queries.rb @@ -0,0 +1,34 @@ +# frozen_string_literal: true + +require_relative '../test_helper' +require_relative '../../lib/medium/985_sum_of_even_numbers_after_queries' +require 'minitest/autorun' + +class SumOfEvenNumbersAfterQueriesTest < ::Minitest::Test + def test_default_one + assert_equal( + [8, 6, 2, 4], + sum_even_after_queries( + [1, 2, 3, 4], + [ + [1, 0], + [-3, 1], + [-4, 0], + [2, 3] + ] + ) + ) + end + + def test_default_two + assert_equal( + [0], + sum_even_after_queries( + [1], + [ + [4, 0] + ] + ) + ) + end +end