백준 알고리즘 8974번 희주의 수학시험(python)

단순 사칙연산 문제이다.

A, B = map(int, input().split())
li = []
t = cnt = 1
for _ in range(B):
    li.append(t)
    if t == cnt:
        cnt = 0
        t += 1
    cnt += 1
print(sum(li[A-1:B]))