二叉树DFS实战：Python刷爆经典面试题

2023-11-21 12:23:06

导语：

对于程序员来说，数据结构是面试中绕不开的话题。而二叉树作为数据结构中的经典，更是备受面试官青睐。今天，我们就用Python这把利器，在十小时内搞定二叉树DFS面试题，让你在面试中大放异彩！

1. DFS简介：深度优先搜索

DFS，即深度优先搜索，是一种以深度为优先级遍历树形结构的算法。它的特点是：一直向下遍历，直到不能再向下遍历为止，再回溯到上一层继续向下遍历。

2. Python实现DFS

在Python中，实现DFS非常简单，只需借助栈数据结构即可：

def dfs(root):
    stack = [root]
    while stack:
        node = stack.pop()
        # 访问节点
        if node.left:
            stack.append(node.left)
        if node.right:
            stack.append(node.right)

3. 经典面试题实战

3.1 求二叉树最大深度

def max_depth(root):
    if not root:
        return 0
    left_depth = max_depth(root.left)
    right_depth = max_depth(root.right)
    return max(left_depth, right_depth) + 1

3.2 判断二叉树是否为平衡树

def is_balanced(root):
    if not root:
        return True
    left_depth = max_depth(root.left)
    right_depth = max_depth(root.right)
    return abs(left_depth - right_depth) <= 1 and is_balanced(root.left) and is_balanced(root.right)

3.3 二叉树路径和

def has_path_sum(root, target_sum):
    if not root:
        return False
    if root.val == target_sum and not root.left and not root.right:
        return True
    return has_path_sum(root.left, target_sum - root.val) or has_path_sum(root.right, target_sum - root.val)