【Leetcode题解】2551. Put Marbles in Bags
Posted on Mon 31 March 2025 in Leetcode
首先我们来看看这个问题,这道题目 首先提供给我们一个弹珠列表weights
,weights[i]
代表了弹珠i
的重量。要求我们将每个弹珠分配到k
个袋子中,并满足如下条件:
- 每个袋子必须有一粒弹珠。
- 每个袋子中包含的弹珠是弹珠列表中的一个连续子数组。即
满足这个条件的袋子的得分为此袋子代表的连续子数组的第一个和最后一个弹珠的 怎么理解呢?如果当前袋子包含weights[i:j+1]
,那么当前袋子的得分为weights[i] + weights[j]
。求最大化和最小化得分之差。
按照题目的意思,若我们将弹珠分配到k
个袋子中,那么实际上我们是在这个数组中放下k-1
分割点,才能将这个数组分为k
个子数组。
考虑一个例子:若weights = [a, b, c, d, e, f, g], k = 3
,考虑两种分法:
P = [a, b], [c, d], [e, f, g]
Q = [a, b, c], [d, e], [f, g]
那么这两种方法各自的得分为score(P) = (a + b) + (c + d) + (e + g)
,score(Q) = (a + c) + (d + e) + (f + g)
。 在以上的式子中,我们使用的方式是按照题目的定义,将每个子数组的头和尾两个数字加起来用括号进行划分。 我们可以观察到在任何一种分法里面,数组的头元素和尾元素永远存在。 那么当我们实际在求结果要求的两种分法的差的时候,数组的头元素和尾元素都是消掉的,对最终的数字影响不计。令score'(P) = score(P) - a - g
和score'(Q) = score(Q) - a - g
,我们可以得到score'(P) = b + c + d + e
以及score'(Q) = c + d + e + f
。 那么,可知score'
的定义为最终得分为每个分割点左右两边的元素之和,且两种分法之差为score'(P) - score'(Q)
。
若weights
有w
个元素,我们有w - 1
个可选的分割点。 我们将这些分割点都求出来并进行排序,就可以拿到令得分最大和最小的分割点了。
class Solution:
def putMarbles(self, weights: List[int], k: int) -> int:
w = [weights[i] + weights[i + 1] for i in range(len(weights) - 1)]
w.sort()
return sum(w[len(w) - k + 1:]) - sum(w[0:k-1])