二叉树的深度优先搜索(DFS)、广度优先搜索(BFS)

20 篇文章 2 订阅

1. 基本概念

在这里插入图片描述
如图所示,BFS为是从根节点开始,沿着树的宽度遍历树的节点。如果所有节点均被访问,则算法中止。如上图所示的二叉树,A 是第一个访问的,然后顺序是 B、C,然后再是 D、E、F、G。

2. 基本思路

仔细看看层序遍历过程,其实就是从上到下,从左到右依次将每个数放入到队列中,然后按顺序依次打印就是想要的结果。所以这里我们借助一个队列进行操作。

实现过程

  1. 首先将二叉树的根节点push到队列中,判断队列不为nullptr,就输出队头的元素,
  2. 判断节点如果有孩子,就将孩子push到队列中,
  3. 遍历过的节点出队列,
  4. 循环以上操作,直到Node == nullptr。

3. 代码

#include <iostream>
#include <queue>

struct Node {
    int value;
    Node* left;
    Node* right;
    Node(int value):
        value(value), left(nullptr), right(nullptr) {}
};

void dfs(Node* head) {
    if (head == nullptr) {
        return;
    }
    std::cout << head->value << ",";
    dfs(head->left);
    dfs(head->right);
}

void bfs(Node* head) {
    if (head == nullptr) {  // if head is nullptr, return directly
        return;
    }
    std::queue<Node*> qbfs;
    qbfs.push(head);
    while (!qbfs.empty()) {
        std::cout << qbfs.front()->value << ",";
        if (qbfs.front()->left != nullptr) {    // if current queue front has left child
            qbfs.push(qbfs.front()->left);
        }
        if (qbfs.front()->right != nullptr) {   // if current queue front has right child
            qbfs.push(qbfs.front()->right);
        }
        qbfs.pop();  // pop if we have alreay visit the node
    }
}
int main() {
    Node* head = new Node(1);
    head->left = new Node(2);
    head->right = new Node(3);
    head->left->left = new Node(4);
    head->left->right = new Node(5);
    head->right->left = new Node(6);
    head->right->right = new Node(7);

    std::cout << "==============DFS==============\n";
    dfs(head);
    std::cout << "\n==============BFS==============\n";
    bfs(head);
    return 0;
}

输出:

    std::cout << "==============DFS==============\n";
    1,2,4,5,3,6,7
    std::cout << "\n==============BFS==============\n";
    1,2,3,4,5,6,7

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
二叉树的广度优先遍历(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 ``` 这样,你就可以使用这两个函数来实现二叉树的广度优先遍历和深度优先遍历了。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值