mirror of
https://github.com/krahets/hello-algo.git
synced 2024-12-26 21:46:29 +08:00
5f814d6538
Fix the figure of tail recursion. Fix two links.
69 lines
1.5 KiB
Python
69 lines
1.5 KiB
Python
"""
|
|
File: recursion.py
|
|
Created Time: 2023-08-24
|
|
Author: Krahets (krahets@163.com)
|
|
"""
|
|
|
|
|
|
def recur(n: int) -> int:
|
|
"""递归"""
|
|
# 终止条件
|
|
if n == 1:
|
|
return 1
|
|
# 递:递归调用
|
|
res = recur(n - 1)
|
|
# 归:返回结果
|
|
return n + res
|
|
|
|
|
|
def for_loop_recur(n: int) -> int:
|
|
"""使用迭代模拟递归"""
|
|
# 使用一个显式的栈来模拟系统调用栈
|
|
stack = []
|
|
res = 0
|
|
# 递:递归调用
|
|
for i in range(n, 0, -1):
|
|
# 通过“入栈操作”模拟“递”
|
|
stack.append(i)
|
|
# 归:返回结果
|
|
while stack:
|
|
# 通过“出栈操作”模拟“归”
|
|
res += stack.pop()
|
|
# res = 1+2+3+...+n
|
|
return res
|
|
|
|
|
|
def tail_recur(n, res):
|
|
"""尾递归"""
|
|
# 终止条件
|
|
if n == 0:
|
|
return res
|
|
# 尾递归调用
|
|
return tail_recur(n - 1, res + n)
|
|
|
|
|
|
def fib(n: int) -> int:
|
|
"""斐波那契数列:递归"""
|
|
# 终止条件 f(1) = 0, f(2) = 1
|
|
if n == 1 or n == 2:
|
|
return n - 1
|
|
# 递归调用 f(n) = f(n-1) + f(n-2)
|
|
res = fib(n - 1) + fib(n - 2)
|
|
# 返回结果 f(n)
|
|
return res
|
|
|
|
|
|
"""Driver Code"""
|
|
if __name__ == "__main__":
|
|
n = 5
|
|
res = recur(n)
|
|
print(f"\n递归函数的求和结果 res = {res}")
|
|
|
|
res = for_loop_recur(n)
|
|
print(f"\n使用迭代模拟递归求和结果 res = {res}")
|
|
|
|
res = tail_recur(n, 0)
|
|
print(f"\n尾递归函数的求和结果 res = {res}")
|
|
|
|
res = fib(n)
|
|
print(f"\n斐波那契数列的第 {n} 项为 {res}")
|