C++实现二叉树深度优先搜索(DFS)、广度优先搜索(BFS)

在前面的文章中我们实现了二叉树的深度优先搜索,先序遍历,中序遍历,后序遍历,分别实现了递归和非递归版本。这里我们着重介绍一下广度优先搜索(BFS)

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;
}

在这里插入图片描述

4. 参考文章

  1. 二叉树 深度优先搜索(DFS)、广度优先搜索(BFS)
  2. 二叉树的层序遍历(两种方法实现)
  3. C++实现二叉树的先序、中序、后序遍历,包括递归方式和非递归 方式
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值