Submission #4033412

---

Source Code Expand

from operator import itemgetter
import numpy as np

N = int(input())
sum=0
sum2=0
count=0
count2=0


s = input().split()

for i in range(N):
    c = int(s[i])
    if i==0:
        sum=c
        count2=abs(c)+1
        sum2=np.sign(c)*(-1)
    else:
        if sum*(sum+c)>=0:
            count += abs(sum+c)+1
            sum = -1 * np.sign(sum)
        else:
            sum+=c

        if sum2 * (sum2 + c) >= 0:
            count2 += abs(sum2 + c) + 1
            sum2 = -1 * np.sign(sum2)
        else:
            sum2+=c

print(min(count,count2))

Submission Info

Judge Result