golang数据结构篇之二叉树

简介: golang数据结构篇之二叉树

二叉树遍历

前序遍历:先访问根节点,再前序遍历左子树,再前序遍历右子树

中序遍历:先中序遍历左子树,再访问根节点,再中序遍历右子树

后序遍历:先后序遍历左子树,再后序遍历右子树,再访问根节点

  • 以根访问顺序决定是什么遍历
  • 左子树都是优先右子树

递归遍历

前中后序递归遍历换换位置即可

func preorderTraversal(root *TreeNode) {
  if root == nil {
    return
  }
  // 先访问根再访问左右
  fmt.Println(root.Val)
  preorderTraversal(root.Left)
  preorderTraversal(root.Right)
}

前序非递归

func preorderTraversal(root *TreeNode) []int {
  // 非递归
  if root == nil{
    return nil
  }
  result:=make([]int,0)
  stack:=make([]*TreeNode,0)
  for root!=nil || len(stack)!=0{
    for root !=nil{
      // 前序遍历,所以先保存结果
      result=append(result,root.Val)
      stack=append(stack,root)
      root=root.Left
    }
    // pop
    node:=stack[len(stack)-1]
    stack=stack[:len(stack)-1]
    root=node.Right
  }
  return result
}

中序非递归

// 思路:通过stack 保存已经访问的元素,用于原路返回
func inorderTraversal(root *TreeNode) []int {
    result := make([]int, 0)
    if root == nil {
        return result
    }
    stack := make([]*TreeNode, 0)
    for len(stack) > 0 || root != nil {
        for root != nil {
            stack = append(stack, root)
            root = root.Left // 一直向左
        }
        // 弹出
        val := stack[len(stack)-1]
        stack = stack[:len(stack)-1]
        result = append(result, val.Val)
        root = val.Right
    }
    return result
}

后续非递归

核心就是:根节点必须在右节点弹出之后,再弹出

func postorderTraversal(root *TreeNode) []int {
    // 通过lastVisit标识右子节点是否已经弹出
    if root == nil {
        return nil
    }
    result := make([]int, 0)
    stack := make([]*TreeNode, 0)
    var lastVisit *TreeNode
    for root != nil || len(stack) != 0 {
        for root != nil {
            stack = append(stack, root)
            root = root.Left
        }
        // 这里先看看,先不弹出
        node:= stack[len(stack)-1]
        // 根节点必须在右节点弹出之后,再弹出
        if node.Right == nil || node.Right == lastVisit {
            stack = stack[:len(stack)-1] // pop
            result = append(result, node.Val)
            // 标记当前这个节点已经弹出过
            lastVisit = node
        } else {
            root = node.Right
        }
    }
    return result
}

DFS 深搜-从上到下

func preorderTraversal(root *TreeNode) []int {
  result := make([]int, 0)
  dfs(root, &result)
  return result
}
func dfs(root *TreeNode, result *[]int) {
  if root == nil {
    return
  }
  *result = append(*result, root.Val)
  dfs(root.Left, result)
  dfs(root.Right, result)
}

DFS 深搜-从下向上(分治法)

func preorderTraversal(root *TreeNode) []int {
  result := divideAndConquer(root)
  return result
}
func divideAndConquer(root *TreeNode) []int {
  result := make([]int, 0)
  // 返回条件(null & leaf)
  if root == nil {
    return result
  }
  // 分治(Divide)
  left := divideAndConquer(root.Left)
  right := divideAndConquer(root.Right)
  // 合并结果(Conquer)
  result = append(result, root.Val)
  result = append(result, left...)
  result = append(result, right...)
  return result
}

DFS 深度搜索(从上到下) 和分治法区别:前者一般将最终结果通过指针参数传入,后者一般递归返回结果最后合并

BFS 层次遍历

func levelOrder(root *TreeNode) [][]int {
    // 通过上一层的长度确定下一层的元素
    result := make([][]int, 0)
    if root == nil {
        return result
    }
    queue := make([]*TreeNode, 0)
    queue = append(queue, root)
    for len(queue) > 0 {
        list := make([]int, 0)
        // 为什么要取length?
        // 记录当前层有多少元素(遍历当前层,再添加下一层)
        l := len(queue)
        for i := 0; i < l; i++ {
            // 出队列
            level := queue[0]
            queue = queue[1:]
            list = append(list, level.Val)
            if level.Left != nil {
                queue = append(queue, level.Left)
            }
            if level.Right != nil {
                queue = append(queue, level.Right)
            }
        }
        result = append(result, list)
    }
    return result
}



目录
相关文章
|
20天前
|
存储 算法 关系型数据库
数据结构与算法学习二一:多路查找树、二叉树与B树、2-3树、B+树、B*树。(本章为了解基本知识即可,不做代码学习)
这篇文章主要介绍了多路查找树的基本概念,包括二叉树的局限性、多叉树的优化、B树及其变体(如2-3树、B+树、B*树)的特点和应用,旨在帮助读者理解这些数据结构在文件系统和数据库系统中的重要性和效率。
14 0
数据结构与算法学习二一:多路查找树、二叉树与B树、2-3树、B+树、B*树。(本章为了解基本知识即可,不做代码学习)
|
20天前
|
存储 算法 搜索推荐
数据结构与算法学习十七:顺序储存二叉树、线索化二叉树
这篇文章主要介绍了顺序存储二叉树和线索化二叉树的概念、特点、实现方式以及应用场景。
15 0
数据结构与算法学习十七:顺序储存二叉树、线索化二叉树
|
22天前
|
Java
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(二)
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(二)
24 1
|
22天前
|
算法 Java C语言
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(一)
【用Java学习数据结构系列】震惊,二叉树原来是要这么学习的(一)
22 1
|
24天前
|
存储
【数据结构】二叉树链式结构——感受递归的暴力美学
【数据结构】二叉树链式结构——感受递归的暴力美学
|
27天前
|
存储 算法 调度
数据结构--二叉树的顺序实现(堆实现)
数据结构--二叉树的顺序实现(堆实现)
|
20天前
|
存储 算法
探索数据结构:分支的世界之二叉树与堆
探索数据结构:分支的世界之二叉树与堆
|
20天前
|
存储 算法
数据结构与算法学习十六:树的知识、二叉树、二叉树的遍历(前序、中序、后序、层次)、二叉树的查找(前序、中序、后序、层次)、二叉树的删除
这篇文章主要介绍了树和二叉树的基础知识,包括树的存储方式、二叉树的定义、遍历方法(前序、中序、后序、层次遍历),以及二叉树的查找和删除操作。
16 0
|
24天前
|
存储 算法
【数据结构】二叉树——顺序结构——堆及其实现
【数据结构】二叉树——顺序结构——堆及其实现
|
26天前
|
存储
【数据结构】二叉树零基础无压力上手,超详解
【数据结构】二叉树零基础无压力上手,超详解
24 0