fork download
  1. # your code goes here
  2. nums = [1,1,1,1,2,2]
  3. k1 = 1
  4. k2 = 3
  5. c1 = 0
  6. c2 = 0
  7. ans = 0
  8. for j in range(1,len(nums)-2):
  9. for i in range(0,j-1):
  10. if(nums[i]+nums[j]>k1):
  11. c1 += 1
  12. k = j+1
  13. l = len(nums)-1
  14. while(l>k):
  15. if(nums[k]+nums[l]>k2):
  16. c2 += 1
  17. l -= 1
  18. else:
  19. k += 1
  20. ans += c1*c2
  21. print(ans)
Success #stdin #stdout 0.08s 7924KB
stdin
Standard input is empty
stdout
11