[Project Euler Problem 6]Sum square difference

def a(x):
    rst = 0
    for i in range(x+1):
        rst += i**2
    return rst

def b(x):
    rst = 0
    for i in range(x+1):
        rst += i
    return rst**2

r = 100
print b(r) - a(r)

.