(通俗易懂)《面试》-- 广度优先搜索(BFS)与深度优先搜索(DFS)Python实现

前言

这两种都是作为经典的图搜索算法,目的都是系统地展开并检查图中所有的节点,但是这两种算法有着不同的搜索方式,以搜索二叉树为例(二叉树其实也是一种特殊的图结构),广度优先搜索(BFS)从上往下搜索只有等待上一层所有节点都遍历完毕才会转到下一层,而深度优先搜索(DFS)则是开始就一直往深处走,知道找到解或者走到底部为止。

广度优先搜索(BFS)

BFS思想(以二叉树为例子)

1、从根节点开始;

2、将节点保存到队列(queue)

3、取出队列中首位节点,读取该节点的值和判断是否有孩子节点,若有则将孩子节点依次添加到队列尾部

4、重复3步骤,直到队列为空

# 定义树节点的类
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left = None
        self.right = None


def BFS(node):
    # 广度优先算法使用队列
    if node is None:
        return None

    queue = [node]
    result = []

    while len(queue):
        cnt = queue.pop(0)        # 取出队列中首位节点
        result.append(cnt.val)    # 读取该节点的值

        if cnt.left:              # 
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
二叉树的广度优先遍历(BFS)可以使用队列来实现,具体步骤如下: 1. 创建一个空队列,并将根节点入队。 2. 当队列不为空时,执行以下操作: - 出队一个节点,将其值存储或打印。 - 将出队节点的左子节点入队(如果存在)。 - 将出队节点的右子节点入队(如果存在)。 3. 重复步骤2,直到队列为空。 下面是一个Python实现的示例代码: ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right def bfs(root): if not root: return [] result = [] queue = [root] while queue: node = queue.pop(0) result.append(node.val) if node.left: queue.append(node.left) if node.right: queue.append(node.right) return result ``` 对于深度优先遍历(DFS),有两种常用的方法:前序遍历、中序遍历和后序遍历。下面以前序遍历为例进行解释。 前序遍历的顺序是:根节点 -> 左子树 -> 右子树。具体实现步骤如下: 1. 创建一个空列表,用于存储遍历结果。 2. 定义一个辅助函数,接收一个节点作为参数: - 若节点为空,返回。 - 将节点的值存储或打印。 - 递归调用辅助函数遍历节点的左子树。 - 递归调用辅助函数遍历节点的右子树。 3. 调用辅助函数,将根节点作为参数传入。 下面是一个Python实现的示例代码: ```python def dfs(root): if not root: return [] result = [] def helper(node): if not node: return result.append(node.val) helper(node.left) helper(node.right) helper(root) return result ``` 这样,你就可以使用这两个函数来实现二叉树的广度优先遍历和深度优先遍历了。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值