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. i = 0
  10. c1 = 0
  11. c2 = 0
  12. while(i<j):
  13. if(nums[i]+nums[j]>k1):
  14. c1 += 1
  15. i += 1
  16. k = j+1
  17. l = len(nums)-1
  18. while(l>k):
  19. if(nums[k]+nums[l]>k2):
  20. c2 += l-k
  21. l -= 1
  22. else:
  23. k += 1
  24. ans += c1*c2
  25. print(ans)# your code goes here
Success #stdin #stdout 0.07s 14212KB
stdin
Standard input is empty
stdout
6