0904-fruit-into-baskets.rb

# frozen_string_literal: true

# 904. Fruit Into Baskets
# https://leetcode.com/problems/fruit-into-baskets

# @param {Integer[]} fruits
# @return {Integer}
deftotal_fruit(fruits)
left=0
max_fruit=0
hash={}
hash.default=0

fruits.each_with_indexdo |fruit,right|
hash[fruit] += 1

whilehash.length > 2
hash[fruits[left]] -= 1

hash.delete(fruits[left])ifhash[fruits[left]].zero?

left += 1
end

max_fruit=[max_fruit,right - left + 1].max
end

max_fruit
end

# ********************#
# TEST #
# ********************#

require"test/unit"
classTest_total_fruit < Test::Unit::TestCase
deftest_
assert_equal3,total_fruit([1,2,1])
assert_equal3,total_fruit([0,1,2,2])
assert_equal4,total_fruit([1,2,3,2,2])
end
end