def fn(arr,d):
s=0
n=len(arr)
for i in range(n):
ts=0
curr=0
for j in range(i,n):
if j==i:
ts+=arr[j]
curr+=arr[j]
elif arr[j]-arr[j-1]==d:
ts+=(arr[j]+curr)
curr+=arr[j]
else:
break
s+=ts
return s
def main(arr):
suma=0
for x in arr:
suma+=x
return fn(arr,0)+fn(arr,1)-suma
print(main([5,8,8,10,11,12]))
ZGVmIGZuKGFycixkKToKICAgIHM9MAogICAgbj1sZW4oYXJyKQogICAgZm9yIGkgaW4gcmFuZ2Uobik6CiAgICAgICAgdHM9MAogICAgICAgIGN1cnI9MAogICAgICAgIGZvciBqIGluIHJhbmdlKGksbik6CiAgICAgICAgICAgIGlmIGo9PWk6CiAgICAgICAgICAgICAgICB0cys9YXJyW2pdCiAgICAgICAgICAgICAgICBjdXJyKz1hcnJbal0KICAgICAgICAgICAgZWxpZiBhcnJbal0tYXJyW2otMV09PWQ6CiAgICAgICAgICAgICAgICB0cys9KGFycltqXStjdXJyKQogICAgICAgICAgICAgICAgY3Vycis9YXJyW2pdCiAgICAgICAgICAgIGVsc2U6CiAgICAgICAgICAgICAgICBicmVhawogICAgICAgIHMrPXRzCiAgICAgICAgCiAgICByZXR1cm4gcwoKZGVmIG1haW4oYXJyKToKICAgIHN1bWE9MAogICAgZm9yIHggaW4gYXJyOgogICAgICAgIHN1bWErPXgKICAgIAogICAgcmV0dXJuIGZuKGFyciwwKStmbihhcnIsMSktc3VtYQoKcHJpbnQobWFpbihbNSw4LDgsMTAsMTEsMTJdKSkK