Add typing annotations to Python codes. (#411)

codes/python/chapter_computational_complexity/space_complexity.py

@@ -8,52 +8,52 @@ import sys, os.path as osp
 sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__))))
 from modules import *
 
-def function():
+def function() -> int:
     """ 函数 """
     # do something
     return 0
 
-def constant(n):
+def constant(n: int) -> None:
     """ 常数阶 """
     # 常量、变量、对象占用 O(1) 空间
-    a = 0
-    nums = [0] * 10000
+    a: int = 0
+    nums: List[int] = [0] * 10000
     node = ListNode(0)
     # 循环中的变量占用 O(1) 空间
     for _ in range(n):
-        c = 0
+        c: int = 0
     # 循环中的函数占用 O(1) 空间
     for _ in range(n):
         function()
 
-def linear(n):
+def linear(n: int) -> None:
     """ 线性阶 """
     # 长度为 n 的列表占用 O(n) 空间
-    nums = [0] * n
+    nums: List[int] = [0] * n
     # 长度为 n 的哈希表占用 O(n) 空间
-    mapp = {}
+    mapp: Dict = {}
     for i in range(n):
         mapp[i] = str(i)
 
-def linear_recur(n):
+def linear_recur(n: int) -> None:
     """ 线性阶（递归实现） """
     print("递归 n =", n)
     if n == 1: return
     linear_recur(n - 1)
 
-def quadratic(n):
+def quadratic(n: int) -> None:
     """ 平方阶 """
     # 二维列表占用 O(n^2) 空间
-    num_matrix = [[0] * n for _ in range(n)]
+    num_matrix: List[List[int]] = [[0] * n for _ in range(n)]
 
-def quadratic_recur(n):
+def quadratic_recur(n: int) -> int:
     """ 平方阶（递归实现） """
     if n <= 0: return 0
     # 数组 nums 长度为 n, n-1, ..., 2, 1
-    nums = [0] * n
+    nums: List[int] = [0] * n
     return quadratic_recur(n - 1)
 
-def build_tree(n):
+def build_tree(n: int) -> Optional[TreeNode]:
     """ 指数阶（建立满二叉树） """
     if n == 0: return None
     root = TreeNode(0)
@@ -74,5 +74,5 @@ if __name__ == "__main__":
     quadratic(n)
     quadratic_recur(n)
     # 指数阶
-    root = build_tree(n)
+    root: Optional[TreeNode] = build_tree(n)
     print_tree(root)