b558: 求數列第 n 項

題目

b558: 求數列第 n 項

https://zerojudge.tw/ShowProblem?problemid=b558

解答

while True:
    try:
        def f(n,m = 1,a = 1):
            return m if n == 1 else f(n-1,m+a,a+1)
        print(f(int(input())))
    except:
        break