Leetcode 3478. Choose K Elements With Maximum Sum

• Leetcode 3478. Choose K Elements With Maximum Sum
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3478. Choose K Elements With Maximum Sum

1. 解题思路

这一题思路上就是一个有序数组，我们首先将数组1有序排列，然后依次考察其每一个位置上的元素，此时就可以保证每一个位置上的元素被考察时，此前数组2当中对应位置的元素都是可用的，我们只需要取出topk个元素进行求和就行了。

当然，由于只需要考虑topk，因此事实上我们只需要维护最大的topk个元素及其对应的和的值即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def findMaxSum(self, nums1: List[int], nums2: List[int], k: int) -> List[int]:
        n = len(nums1)
        ans = [0 for _ in range(n)]
        ordered_nums1 = sorted([(x, i) for i, x in enumerate(nums1)])
        pre_max, topk_sum = 0, 0
        cache, topk_elems = [], []
        
        for num, idx in ordered_nums1:
            if num > pre_max:
                for candidate in cache:
                    if len(topk_elems) < k:
                        bisect.insort(topk_elems, candidate)
                        topk_sum += candidate
                    elif topk_elems[0] < candidate:
                        bisect.insort(topk_elems, candidate)
                        topk_sum += candidate - topk_elems[0]
                        topk_elems.pop(0)
                cache = []
                pre_max = num
            
            ans[idx] = topk_sum
            cache.append(nums2[idx])
        return ans

提交代码评测得到：耗时1165ms，占用内存48.3MB。