LeetCode in Kotlin

2144. Minimum Cost of Buying Candies With Discount

Easy

A shop is selling candies at a discount. For every two candies sold, the shop gives a third candy for free.

The customer can choose any candy to take away for free as long as the cost of the chosen candy is less than or equal to the minimum cost of the two candies bought.

Given a 0-indexed integer array cost, where cost[i] denotes the cost of the ith candy, return the minimum cost of buying all the candies.

Example 1:

Input: cost = [1,2,3]

Output: 5

Explanation: We buy the candies with costs 2 and 3, and take the candy with cost 1 for free.

The total cost of buying all candies is 2 + 3 = 5. This is the only way we can buy the candies.

Note that we cannot buy candies with costs 1 and 3, and then take the candy with cost 2 for free.

The cost of the free candy has to be less than or equal to the minimum cost of the purchased candies.

Example 2:

Input: cost = [6,5,7,9,2,2]

Output: 23

Explanation: The way in which we can get the minimum cost is described below:

Hence, the minimum cost to buy all candies is 9 + 7 + 5 + 2 = 23.

Example 3:

Input: cost = [5,5]

Output: 10

Explanation: Since there are only 2 candies, we buy both of them. There is not a third candy we can take for free.

Hence, the minimum cost to buy all candies is 5 + 5 = 10.

Constraints:

Solution

class Solution {
    fun minimumCost(cost: IntArray): Int {
        cost.sort()
        var size = 0
        var sum = 0
        for (i in cost.indices.reversed()) {
            size++
            if (size % 3 != 0) {
                sum += cost[i]
            }
        }
        return sum
    }
}