函数main()接受一个大于等于1的正整数n和一个介于0,9的正整数a,要求返回表达式一个a➕两个a➕到n个a的前n项和。(python)不用循环,要求用lambda和map
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!")