# Definition for a binary tree node. # class TreeNode: # def __init__(self, val=0, left=None, right=None): # self.val = val # self.left = left # self.right = right classSolution: definorderTraversal(self, root: Optional[TreeNode]) -> List[int]: ans = [] defdfs(node): if node == None: return dfs(node.left) ans.append(node.val) dfs(node.right) dfs(root) return ans
复杂度分析
时间复杂度:O(N),其中 N 是二叉树的节点数。每个节点被访问且处理且仅处理一次。
空间复杂度:O(H),其中 H 是树的高度。主要由递归调用的栈空间产生。平均情况下高度平衡的树空间为 O(log N),最坏情况(退化为链状)下为 O(N)。
classTreeNode: def__init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right
defbuild_tree(arr): ifnot arr or arr[0] isNone: returnNone root = TreeNode(arr[0]) queue = [root] i = 1 while queue and i < len(arr): node = queue.pop(0) if arr[i] isnotNone: node.left = TreeNode(arr[i]) queue.append(node.left) i += 1 if i < len(arr) and arr[i] isnotNone: node.right = TreeNode(arr[i]) queue.append(node.right) i += 1 return root
deftree_to_list(root): ifnot root: return [] result = [] queue = [root] while queue: node = queue.pop(0) if node: result.append(node.val) queue.append(node.left) queue.append(node.right) else: result.append(None) while result and result[-1] isNone: result.pop() return result
classSolution: definorderTraversal(self, root: Optional[TreeNode]) -> List[int]: ans = [] defdfs(node): if node == None: return dfs(node.left) ans.append(node.val) dfs(node.right) dfs(root) return ans
defmain(): arr = sys.stdin.read().strip().split() tree_arr = [Noneif v == 'None'elseint(v) for v in arr] root = build_tree(tree_arr)
sol = Solution() result = sol.inorderTraversal(root) print(result)