#94. 二叉树的中序遍历

#94. 二叉树的中序遍历 (递归与迭代)

题目

给定一个二叉树，返回中序遍历结果。（左子树，中间节点，右子树）

解答

递归解法

• base case：如果为None，则返回 []
• recursive case: 返回inorder(左) + [中] + inorder(右)

class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        if not root:
            return []
        return self.inorderTraversal(root.left) + [root.val] + self.inorderTraversal(root.right)

###迭代解法

• 设置curr为root，不断更新curr为curr.left 并把curr加入stack中，直到curr不存在。
• 如果curr不存在，则pop栈顶的node为curr，把其value添加入result中，并更新curr为curr.right。
• 重复上述操作直到stack 空，curr 为None。

class Solution:
    def inorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
        stack = []
        result = []
        
        curr = root
        while stack or curr:
            if curr:
                stack.append(curr)
                curr = curr.left
            else:
                curr = stack.pop()
                result.append(curr.val)
                curr = curr.right
        return result

收获

• 关键点在于，每一个新入stack的节点代表着上一级的node节点，当下一级的左侧为空时，就会返回到上一级节点，输出它的value，把curr设成上一级节点的右侧节点：再从右子树开始溯源。