Question

我正在使用以下程序查找最大和和的索引。我能够获得正确的索引，但是在找到正确的索引时遇到了困难。

def max_sum_new(a):
max_current = max_global = a[0]
r_index = 0
for i in range(1, len(a)):
    max_current = max(a[i], max_current+a[i])

    if max_current > max_global:
        max_global = max_current
        r_index = i

return (max_global, r_index)

#Driver Code:
my_arr = [-2, 3, 2, -1]
print(max_sum_new(my_arr))

我得到了

(5, 2)

这是预期的，但我也想获取起始索引，在这种情况下，该索引应为1

所以我希望

(5, 1, 2)

有什么办法可以在这里获得开始索引吗？我应该如何将变量记录到开始索引中？

Answer 1

您只需要应用与r_index相同的逻辑即可。每当您更改max_current时，即最大子数组的开始更改。最后看起来像这样

max_current = max_global = a[0]
r_index = 0
left_index = 0
for i in range(1, len(a)):
    if a[i] > max_current + a[i]:
        # Here is where your start index changes
        left_index = i
        max_current = a[i]
    else:
        max_current = a[i] + max_current

    if max_current > max_global:
        max_global = max_current
        r_index = i

print(max_global, left_index, r_index)

如何获得最大和子数组的起始索引

1 个答案: