def func(n):
    res = 0
    # 第一段双层循环
    for i in range(n):
        for j in range(n):
            res += i * j
    # 第二段单层循环
    for k in range(n):
        res += k
    return res