计算与 K 之和大于最大元素的元素

在一个已排序的整数数组中，我们需要找到所有计算某个数K与该数后面的数之和大于该数最大值的数。我们需要返回这些元素的下标，其中，被返回的下标需要按升序排序。如果有多个结果，我们需要返回所有结果。

例如：

输入: [2, 3, 4, 5, 6, 7, 8], K = 10 输出: [2, 3, 4]

思路

我们可以用双指针的方法来解决这个问题。我们用一个指针start指向开头，一个指针end指向结尾，分别计算start到end之间数的和和最大值，判断是否符合题目要求。

如果当前的和小于等于最大值，我们将end指针向右移动，将end指向下一个元素。如果当前的和大于最大值，我们将start指针向右移动，将start指向下一个元素。当start和end指针相遇时，我们结束查找。

代码

def max_sum_elements(nums, k):
    res = []
    start = 0
    end = 1
    length = len(nums)
    while start < length:
        cur_sum = nums[start]
        while end < length and cur_sum + nums[end] <= nums[length - 1] and cur_sum + nums[end] <= k:
            cur_sum += nums[end]
            end += 1
        if cur_sum + nums[end-1] > nums[length - 1] and cur_sum + nums[end-1] > k:
            res.append(start)
        start += 1
        end = start + 1
    return res

测试

对于输入[2, 3, 4, 5, 6, 7, 8], K = 10，程序的输出为[2,3,4]。

我们可以进一步测试程序，输入[1,2,3,4,5,6,7,8,9,10], K = 20，期望的输出是[0, 1, 2, 3, 4, 5, 6, 7, 8]。