用递归超过了范围。咋办

Question

函数main()接受一个大于等于1的正整数n和一个介于0，9的正整数a，要求返回表达式一个a➕两个a➕到n个a的前n项和。(python)不用循环，要求用lambda和map

Answer 1

main函数如下：

def main(n, a):
    return sum(map(lambda x: x*a, list(range(n+1))))

测试：

def main(n, a):
    return sum(map(lambda x: x*a, list(range(n+1))))

if __name__ == "__main__": #test
    assert 1*4+2*4 == main(2, 4)
    assert 1*6+2*6+3*6 == main(3, 6)

    print("OK!")