import sys (t,r) = (int(sys.argv[1]), int(sys.argv[2])) print( # i : time of charging, between 0 and t # v = i # speed in m/s # d = v * (t-i) # distance sum(i*(t-i) > r for i in range(t)) )