树的深度优先(DFS)和广度优先(BFS)算法

引言

DFS

深度优先算法:从一个节点开始,一直搜索,直到一个叶子节点或者无法搜索下去。

BFS

广度优先算法(层序遍历):从一个节点开始,逐层搜索节点。

实现

DFS(深度优先)

python 复制代码
class Solution:
    def connect(self, root: 'Node') -> 'Node':
        pre = []
        def dfs(node: 'Node', depth: int) -> None:
            if node is None:
                return
            if depth == len(pre): 
                pre.append(node)
            else:  
                pre[depth].next = node  
                pre[depth] = node
            dfs(node.left, depth + 1)
            dfs(node.right, depth + 1)
        dfs(root, 0)  
        return root

这个主要是通过一直遍历左边子树的左节点,核心思想是pre[depth].next = node pre[depth] = node,这个就直接使用将左子树的左节点作为下一个值,直到后面没有值了,这里面的一个判断if depth == len(pre)就做一个是否当前深度等于列表长度。然后就是递归左右子树,进入下一层。

BFS(广度优先/层序遍历)

python 复制代码
class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if root is None:
            return None
        q = [root]
        while q:
            for x, y in pairwise(q):
                x.next = y
            tmp = q
            q = []
            for node in tmp:
                if node.left:  q.append(node.left)
                if node.right: q.append(node.right)
        return root

广度优先搜索和深度优先搜索的主要区别就是广度优先遍历是没有一直往左子树的左节点的,而是直接用pairwise将两个值进行分组,并将下一节点放在分组的另一个值上,for x,y in pairwose(q):

x.next = y,然后将所有的值添加到队列中。对了,队列是一种很适合用于层序遍历的,因为队列是先进先出,而且队列由列表表示。

相关推荐
zhurui_xiaozhuzaizai1 小时前
模型训练-关于token【低概率token, 高熵token】
人工智能·算法·自然语言处理
ThreeYear_s1 小时前
基于FPGA的PID算法学习———实现PID比例控制算法
学习·算法·fpga开发
子豪-中国机器人2 小时前
C++ 信息学奥赛总复习题答案解析
开发语言·c++·算法
弥彦_2 小时前
牛客round95D
c++·算法
强盛小灵通专卖员3 小时前
基于深度学习RT-DETR算法的盲人障碍物目标检测:提升盲人出行安全的智能化突破
深度学习·算法·目标检测·计算机视觉·rt-detr·小论文·计算机期刊
我不是小upper3 小时前
统计学核心概念与现实应用精解(偏机器学习)
算法·机器学习·统计学
Renlijuande3 小时前
百度之星2021——BD202104 萌新
算法
黑牛先生4 小时前
【数据结构】图算法(代码)
数据结构·算法
tt5555555555554 小时前
C++ 标准模板库(STL)详解文档
数据结构·c++·算法
Moonbit4 小时前
MoonBit Pearls Vol.01:使用MoonBit编写Pratt解析器
算法