用 Python 编写程序，查找总和最大的 k 个子列表，并按升序返回总和

pythonserver side programmingprogramming

假设我们有一个名为 nums 的数字列表和另一个值 k，我们必须找到总和最大的 k 个子列表，并以非递减顺序返回总和。

因此，如果输入为 nums = [2, 4, 5, -100, 12, -30, 6, -2, 6] k = 3，则输出将为 [10, 11, 12]，因为我们有这 3 个总和最大的子列表 − [6, -2, 6], [2, 4, 5], [12]。

为了解决这个问题，我们将遵循以下步骤 −

ps := 一个 1 + nums 大小的列表并用 0 填充
对于每个索引 i 和值 v nums，执行
- ps[i + 1] := v + ps[i]
hp := 一个新列表
对于范围从 0 到 ps 大小的 i，执行
- 对于范围从 i + 1 到 ps 大小的 j，执行
  - 将 -(ps[j] - ps[i]) 插入 ps 堆
res := 弹出 ps 堆中的所有元素并反转它们
返回 res

让我们看看下面的实现以便更好地理解 −

示例

from heapq import heappop, heappush
class Solution:
   def solve(self, nums, k):
      ps = [0 for _ in range(len(nums) + 1)]
      for i, v in enumerate(nums):
         ps[i + 1] = v + ps[i]
      hp = []
      for i in range(len(ps)):
         for j in range(i + 1, len(ps)):
            heappush(hp, -(ps[j] - ps[i]))
            return list(reversed([-heappop(hp)
      for _ in range(k)]))
ob = Solution()
nums = [2, 4, 5, -100, 12, -30, 6, -2, 6] k = 3
print(ob.solve(nums, k))