Submission #961003

---

Source Code Expand

import sys
import math

line = sys.stdin.readline()
(N, K) = map(int, line.split())

line = sys.stdin.readline()
a = [i for i in map(int, line.split())]
total = 0
total1 = 0
total2 = 0
total3 = 0

L = N - K
for i in range(0, L):
    total1 += a[i] * (i + 1)

for i in range(N - L, N):
    total3 += a[i] * (N - i)

for i in range(L, N - L):
    total2 += a[i]

total2 *= max(L + 1, 0)

total = 0
total += total1
total += total2
total += total3

print(total)

Submission Info

Judge Result