Python-平面劃分問題

求該題目程式碼

平面上的 n 條直線最多可將平面劃分成幾個區域？ 

輸入說明： 
輸入一個正整數 n ，代表有 n 條直線。 

輸出說明 : 
輸出一個正整數表示有n條直線時，最多能將平面切割成幾個區域。 

結尾不須換行。 

Input: 
------------------------- 
1 
Output: 
------------------------ 
2 

Input: 
------------------------- 
2 
Output: 
------------------------ 
4 

Input: 
------------------------- 
5 
Output: 
------------------------ 
16

最多能將平面切割成（n²+n+2）/ 2 個區域
参考：https://www.zhihu.com/question/448631761


num = int(input('请輸入一個正整數 n ，代表有 n 條直線：'))

def get_area(n):
  return int((n * n + n + 2) / 2)

print(get_area(num))


def f(n):
    if n == 1:
        return 2
    return f(n - 1) + n


if __name__ == '__main__':
    n = int(input())
    res = f(n)
    print(res)

img