Почему мой подход с суммой префиксов дает неправильный результат для некоторых тестовых случаев? [закрыто] ⇐ Python

[Anonymous]

Я пытаюсь использовать суммы префиксов, чтобы быстро найти суммы подмассивов, но мой код дает правильные результаты для некоторых входных данных и неправильные для других:

Код: Выделить всё

nums = [1, 2, 3, 4]
prefix = len(nums) * [0]

for i in range (len(nums)):
prefix[i] = nums[i] + prefix[i-1]

print(prefix)

Я думал, что первый элемент будет 1, но оказалось nums[-1] + 1.
Что я делаю не так и как мне правильно построить массив сумм префиксов?>

Подробнее здесь: https://stackoverflow.com/questions/798 ... test-cases