leetcode 2542. 最大子序列的分数

给你两个下标从 0 开始的整数数组 nums1 和 nums2 ，两者长度都是 n ，再给你一个正整数 k 。你必须从 nums1 中选一个长度为 k 的 子序列 对应的下标。

对于选择的下标 i0 ，i1 ，…， ik - 1 ，你的 分数 定义如下：

nums1 中下标对应元素求和，乘以 nums2 中下标对应元素的 最小值 。
用公示表示： (nums1[i0] + nums1[i1] +…+ nums1[ik - 1]) * min(nums2[i0] , nums2[i1], … ,nums2[ik - 1]) 。
请你返回 最大 可能的分数。

一个数组的 子序列 下标是集合 {0, 1, …, n-1} 中删除若干元素得到的剩余集合，也可以不删除任何元素。

示例 1：

输入：nums1 = [1,3,3,2], nums2 = [2,1,3,4], k = 3
输出：12
解释：
四个可能的子序列分数为：

• 选择下标 0 ，1 和 2 ，得到分数 (1+3+3) * min(2,1,3) = 7 。
• 选择下标 0 ，1 和 3 ，得到分数 (1+3+2) * min(2,1,4) = 6 。
• 选择下标 0 ，2 和 3 ，得到分数 (1+3+2) * min(2,3,4) = 12 。
• 选择下标 1 ，2 和 3 ，得到分数 (3+3+2) * min(1,3,4) = 8 。
  所以最大分数为 12 。
  示例 2：

输入：nums1 = [4,2,3,1,1], nums2 = [7,5,10,9,6], k = 1
输出：30
解释：
选择下标 2 最优：nums1[2] * nums2[2] = 3 * 10 = 30 是最大可能分数。

提示：

n == nums1.length == nums2.length
1 <= n <= 105
0 <= nums1[i], nums2[j] <= 105
1 <= k <= n
原始题目链接

思路，可以把 nums1 和 nums2 组合到一起后按 nums2 从大到小排序，先用最小值堆如堆 k个元素，因为往右 nums2[i] 会逐渐变小，故只有 nums1[i] 大于堆顶元素(sum会变大)，结果才可能变大。

from functools import cmp_to_key
import heapq
class Solution:
    def compare(self, a, b):  
        if a[1] >= b[1]:
            return -1
        else:
            return 1

    def maxScore(self, nums1: List[int], nums2: List[int], k: int) -> int:
        nums = []
        for i in range(len(nums1)):
            nums.append((nums1[i], nums2[i]))
        ### 根据 nums2 从大到小排序
        nums = sorted(nums, key=cmp_to_key(self.compare)) 
        ss = []
        sum, res = 0, 0
        for i in range(k):
            heapq.heappush(ss, nums[i][0])  ## 最小值堆
            sum += nums[i][0]
        res = max(res, sum*nums[k-1][1])
        ## 往右遍历时, 最小值会一直变小, 故只有sum 变大时才可能会更新 res
        for i in range(k, len(nums)):
            if nums[i][0] > ss[0]:
                sum += nums[i][0] - ss[0]
                heapq.heappop(ss)
                heapq.heappush(ss, nums[i][0])
                res = max(res, sum*nums[i][1])
        return res