fork download 
  1. def fn(arr,d):
  2.     cnt=0
  3.     s=0
  4.     curr=0
  5.     for i in range(len(arr)):
  6.         if i==0:
  7.             s=arr[i]
  8.             curr=arr[i]
  9.             cnt=1
  10.         else:
  11.             diff=arr[i]-arr[i-1]
  12.             if diff==d:
  13.                 s+=(arr[i]*(cnt+1)+curr)
  14.                 curr+=arr[i]*(cnt+1)
  15.                 cnt+=1
  16.             else:
  17.                 s+=arr[i]
  18.                 curr=arr[i]
  19.                 cnt=1
  20.  
  21.     return s
  22.  
  23. def main(arr):
  24.     suma=0
  25.     for x in arr:
  26.         suma+=x
  27.  
  28.     return fn(arr,0)+fn(arr,1)-suma
  29.  
  30.  
  31. arr=[5,8,8,10,11,12]
  32. print(main(arr))
  33.  
Success #stdin #stdout 0.07s 14080KB
comments (?)
stdin 
Standard input is empty
stdout 
147
https://ideone.com/gVq0ys
language:
Python 3 (python  3.12)
created:
6 hours ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!