fork download
  1. def fn(arr,d):
  2. s=0
  3. n=len(arr)
  4. for i in range(n):
  5. ts=0
  6. curr=0
  7. for j in range(i,n):
  8. if j==i:
  9. ts+=arr[j]
  10. curr+=arr[j]
  11. elif arr[j]-arr[j-1]==d:
  12. ts+=(arr[j]+curr)
  13. curr+=arr[j]
  14. else:
  15. break
  16. s+=ts
  17.  
  18. return s
  19.  
  20. def main(arr):
  21. suma=0
  22. for x in arr:
  23. suma+=x
  24.  
  25. return fn(arr,0)+fn(arr,1)-suma
  26.  
  27. print(main([5,8,8,10,11,12]))
  28.  
Success #stdin #stdout 0.06s 14188KB
stdin
Standard input is empty
stdout
147