028.py

# time cost = 1.47 µs ± 2.67 ns

def main(n=500):
    res = 16*n**3/3 + 10*n**2 + 26*n/3 + 1
    return int(res)