From 1308a24afe1f12ff68186707fac3f2828e214c2a Mon Sep 17 00:00:00 2001 From: fartem Date: Fri, 3 May 2024 06:35:50 +0300 Subject: [PATCH] 2024-05-03 v. 5.6.0: added "2367. Number of Arithmetic Triplets" --- README.md | 1 + leetcode-ruby.gemspec | 2 +- lib/easy/2367_number_of_arithmetic_triplets.rb | 12 ++++++++++++ test/easy/test_2367_number_of_arithmetic_triplets.rb | 12 ++++++++++++ 4 files changed, 26 insertions(+), 1 deletion(-) create mode 100644 lib/easy/2367_number_of_arithmetic_triplets.rb create mode 100644 test/easy/test_2367_number_of_arithmetic_triplets.rb diff --git a/README.md b/README.md index f98a4a94..3dc416b3 100644 --- a/README.md +++ b/README.md @@ -420,3 +420,4 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/). | 2351. First Letter to Appear Twice | [Link](https://leetcode.com/problems/first-letter-to-appear-twice/) | [Link](./lib/easy/2351_first_letter_to_appear_twice.rb) | | 2357. Make Array Zero by Subtracting Equal Amounts | [Link](https://leetcode.com/problems/make-array-zero-by-subtracting-equal-amounts/) | [Link](./lib/easy/2357_make_array_zero_by_subtracting_equal_amounts.rb) | | 2363. Merge Similar Items | [Link](https://leetcode.com/problems/merge-similar-items/) | [Link](./lib/easy/2363_merge_similar_items.rb) | +| 2367. Number of Arithmetic Triplets | [Link](https://leetcode.com/problems/number-of-arithmetic-triplets/) | [Link](./lib/easy/2367_number_of_arithmetic_triplets.rb) | diff --git a/leetcode-ruby.gemspec b/leetcode-ruby.gemspec index 1e52d835..3d2a891b 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 = '5.5.9' + s.version = '5.6.0' s.license = 'MIT' s.files = ::Dir['lib/**/*.rb'] + %w[bin/leetcode-ruby README.md LICENSE] s.executable = 'leetcode-ruby' diff --git a/lib/easy/2367_number_of_arithmetic_triplets.rb b/lib/easy/2367_number_of_arithmetic_triplets.rb new file mode 100644 index 00000000..afb2ff8b --- /dev/null +++ b/lib/easy/2367_number_of_arithmetic_triplets.rb @@ -0,0 +1,12 @@ +# frozen_string_literal: true + +require 'set' + +# https://leetcode.com/problems/number-of-arithmetic-triplets/ +# @param {Integer[]} nums +# @param {Integer} diff +# @return {Integer} +def arithmetic_triplets(nums, diff) + uniq = nums.to_set + nums.uniq.sum { |num| uniq.include?(num + diff) && uniq.include?(num + 2 * diff) ? 1 : 0 } +end diff --git a/test/easy/test_2367_number_of_arithmetic_triplets.rb b/test/easy/test_2367_number_of_arithmetic_triplets.rb new file mode 100644 index 00000000..32e959c5 --- /dev/null +++ b/test/easy/test_2367_number_of_arithmetic_triplets.rb @@ -0,0 +1,12 @@ +# frozen_string_literal: true + +require_relative '../test_helper' +require_relative '../../lib/easy/2367_number_of_arithmetic_triplets' +require 'minitest/autorun' + +class NumberOfArithmeticTripletsTest < ::Minitest::Test + def test_default + assert_equal(2, arithmetic_triplets([0, 1, 4, 6, 7, 10], 3)) + assert_equal(2, arithmetic_triplets([4, 5, 6, 7, 8, 9], 2)) + end +end