Put Marbles in Bags

You have k bags. You are given a 0-indexed integer array weights where weights[i] is the weight of the i^th marble. You are also given the integer k.

Divide the marbles into the k bags according to the following rules:

  • No bag is empty.
  • If the i^th marble and j^th marble are in a bag, then all marbles with an index between the i^th and j^th indices should also be in that same bag.
  • If a bag consists of all the marbles with an index from i to j inclusively, then the cost of the bag is weights[i] + weights[j].

The score after distributing the marbles is the sum of the costs of all the k bags.

Return the difference between the maximum and minimum scores among marble distributions.

Example 1
Inputweights = [1,3,5,1], k = 2
Output4
The minimal score is 6 from [1],[3,5,1], and the maximal score is 10 from [1,3],[5,1], so the difference is 4.
Example 2
Inputweights = [1,3], k = 2
Output0
The only distribution possible is [1],[3], so the maximum and minimum scores are the same.

Constraints

  • 1 <= k <= weights.length <= 10^5
  • 1 <= weights[i] <= 10^9

Asked at 8 companies

</>

Your Solution

(Ctrl/Cmd + Enter)

Switching Language

Loading template...

Loading...

Sign in to save your progress

AI code evaluation

Get a correctness verdict, missed edge cases, and complexity analysis of your solution.

Sign in to evaluate