美文网首页
JS实现二叉树的遍历(DFS、BFS、前中后序遍历)

JS实现二叉树的遍历(DFS、BFS、前中后序遍历)

作者: kugua666 | 来源:发表于2020-08-06 09:27 被阅读0次

对于二叉树,有深度遍历(DFS)和广度遍历(BFS),深度遍历有前序遍历、中序遍历和后序遍历三种方法,广度遍历也叫做层序遍历。深度遍历的思路是用递归或者循环的方式,广度遍历可以用队列这种数据结构。四种遍历的方式分别是:
(1)前序遍历:根节点,左节点,右节点。
(2)中序遍历:左节点,根节点,右节点。
(3)后序遍历:左节点,右节点,根节点。
(4)广度遍历:按每层遍历
JavaScript代码实现如下:
分别写一个遍历函数,并且以根节点作为形参,返回一个由节点值组成的数组。创建节点的构造函数如下:

function TreeNode(x){
    this.val = x
    this.left = null
    this.right = null
}

一、前序遍历
(1)递归的实现,这种方式思路简单,但消耗空间大

let res = []
function preorderTraverse(pRoot){
    if(!pRoot){//递归的出口
        return 
    }
    res.push(pRoot.val)
    preorderTraverse(pRoot.left)
    preorderTraverse(pRoot.right)
}

(2)循环的方式:这里用到栈这种数据结构,为了访问右子树而存在。对每个节点
a、访问其节点值,并把该节点入栈,在将当前节点设置为左子树。
b、判断节点是否为空,若为空,则将栈顶节点出栈,将当前节点设置为右节点;若不为空,则执行a步骤,直到当前节点为空或者栈为空。

function preorderTraverse(pRoot){
    if(!pRoot){ //判断根节点是否为空
        return []
    }
    let res = [], stack = []
    while(pRoot !== null || stack.length !== 0) {
        while(pRoot !== null){
            res.push(pRoot.val) //先访问当前节点值
            stack.push(pRoot) //将当前节点推入栈中,为了便于访问右节点
            pRoot = pRoot.left //将节点设置为左节点,并通过while循环判断是否为空
        }
        pRoot = stack.pop() 
        pRoot = pRoot.right//将节点设置为右节点
    }
    return res
}

二、中序遍历

//递归的方式
let res = []
function inorderTraverse(pRoot){
    if(!pRoot){//递归的出口
        return 
    }
    inorderTraverse(pRoot.left)
    res.push(pRoot.val)
    inorderTraverse(pRoot.right)
}
//循环的方式
function inorderTraverse(pRoot){
    if(!pRoot){
        return []
    }
    let res = [], stack = []
    while(pRoot !== null || stack.length !== 0) {
        while(pRoot !== null){
            stack.push(pRoot)
            pRoot = pRoot.left
        }
        pRoot = stack.pop()
        res.push(pRoot.val)
        pRoot = pRoot.right
    }
    return res
}

三、后序遍历

let res = []
function postorderTraverse(pRoot){
    if(!pRoot){//递归的出口
        return 
    }
    postorderTraverse(pRoot.left)
    postorderTraverse(pRoot.right)
    res.push(pRoot.val)
}

四、广度遍历

function bfs(pRoot){
    if(!node){
        return []
    }
    let queue = [pRoot]  //将根节点加入到队列中
    let result = []  //结果数组
    while(queue.length){
        let len = queue.length   //队列中长度循环一次就得变一次
        let tempArr = []  //存储每层节点值的临时数组,方便一层层打印
        for(let i = 0; i < len; i++){   //遍历每层节点
            let temp = queue.shift()
            tempArr.push(temp.val)
            if(temp.left){
                queue.push(pNode.left)
            }
            if(temp.right){
                queue.push(pNode.right)
            }
        }
       result.push([...tempArr])
    }
    return result
}

相关文章

  • (补)第四周算法备忘

    深度优先, DFS 前中后序遍历二叉树 典型题目 岛屿问题 八皇后问题 广度优先, BFS 层级遍历n叉树 典型题...

  • JS实现二叉树的遍历(DFS、BFS、前中后序遍历)

    对于二叉树,有深度遍历(DFS)和广度遍历(BFS),深度遍历有前序遍历、中序遍历和后序遍历三种方法,广度遍历也叫...

  • 二叉树的遍历

    递归的宗旨: 先序遍历、中序遍历、后序遍历一般使用深度优先搜索DFS实现,层次遍历一般用广度优先搜索BFS实现。 ...

  • 二叉树 基础操作

    二叉树的使用 二叉树结构 先序创建二叉树 DFS 先序遍历二叉树 中序遍历二叉树 后序遍历二叉树 BFS 层次遍历...

  • 题目链接:后序遍历 题目链接:树的直径 bfs: dfs:

  • 算法-二叉树的遍历实现

    简述 二叉树的遍历分 DFS【深度优先遍历】 和 BFS【广度优先遍历】 两类,其中 DFS 又分为前序遍历,中序...

  • 常考算法面试题系列:树的遍历

    首文发布在 个人博客 两种通用的遍历树的策略 DFS(深度优先遍历):先序遍历,中序遍历,后序遍历; BFS(广度...

  • 二叉树遍历/搜索

    遍历 1.宽度优先遍历BFS 利用队列 2.深度优先遍历DFS 1.递归:前序中序后序 2.非递归:前序中序后序 利用栈

  • 【算法训练营学习笔记-Week06】一遍不懂就多刷几遍

    字典树(Trie ) 温故知新: 树的定义 二叉树,前中序列遍历,层次遍历 DFS和BFS 二叉搜索树(BFS)定...

  • 树相关的题目

    二叉树的构建:左子树,跟节点,右子树 二叉树的遍历:前序,中序,后序,DFS,BFS,所有路径 二叉树深度:最大深...

网友评论

      本文标题:JS实现二叉树的遍历(DFS、BFS、前中后序遍历)

      本文链接:https://www.haomeiwen.com/subject/cftzrktx.html