2706-buy-two-chocolates.rb

# frozen_string_literal: true

# 2706. Buy Two Chocolates
# Easy
# https://leetcode.com/problems/buy-two-chocolates

=begin
You are given an integer array prices representing the prices of various chocolates in a store. You are also given a single integer money, which represents your initial amount of money.
You must buy exactly two chocolates in such a way that you still have some non-negative leftover money. You would like to minimize the sum of the prices of the two chocolates you buy.
Return the amount of money you will have leftover after buying the two chocolates. If there is no way for you to buy two chocolates without ending up in debt, return money. Note that the leftover must be non-negative.

Example 1:
Input: prices = [1,2,2], money = 3
Output: 0
Explanation: Purchase the chocolates priced at 1 and 2 units respectively. You will have 3 - 3 = 0 units of money afterwards. Thus, we return 0.

Example 2:
Input: prices = [3,2,3], money = 3
Output: 3
Explanation: You cannot buy 2 chocolates without going in debt, so we return 3.

Constraints:
2 <= prices.length <= 50
1 <= prices[i] <= 100
1 <= money <= 100
=end

# @param {Integer[]} prices
# @param {Integer} money
# @return {Integer}
def buy_choco(prices, money)
  prices.sort!

  if prices[0] + prices[1] <= money
    return money - prices[0] - prices[1]
  end

  money
end

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

require "test/unit"
class Test_buy_choco < Test::Unit::TestCase
  def test_
    assert_equal 0, buy_choco([1, 2, 2], 3)
    assert_equal 3, buy_choco([3, 2, 3], 3)
  end
end