Сила заклинания вася нашел древний свиток с заклинанием. заклинание состоит из n чисел. сила заклинания вычисляется следующим

Question

Информатика: Сила заклинания вася нашел древний свиток с заклинанием. заклинание состоит из n чисел. сила заклинания вычисляется следующим образом: для каждой подстроки (в данном случае это последовательность подряд идущих чисел), считается сумма чисел в этой подстроки, а затем вычисляется сумма всех сумм подстрок. васе определить силу заклинания. формат входных данных в первой строке содержится число n (1 ≤ n ≤ 10000) - количество чисел в записи заклинания. во второй строке содержится n чисел, заклинание. каждое

sanymoskva · Answer

Исходные числа a(0), a(1), ..., a(N - 1).
Пусть f(k - 1) — сумма сумм всех чисел из подстрок, кончающихся на числе с номером k - 1. Вычислим f(k).
f(k) = (a(0) + a(1) + a(2) + ... + a(k - 1) + a(k)) + (a(1) + a(2) + ... + a(k - 1) + a(k)) + (a(2) + ... + a(k - 1) + a(k)) + ... + (a(k - 1) + a(k)) + a(k) = f(k - 1) + (k + 1) * a(k)
Искомая сумма есть сумма всех f(k).

python 3.6:
n = int(input())
a = enumerate(map(int, input().split()))
s = f = 0
for k, ak in a:
f += (k + 1) * ak
s += f
print(s)

Популярные вопросы