二叉树遍历——中序遍历(Golang)

二叉树遍历——中序遍历(Golang)

  • 简介
  • 定义
  • Golang遍历实现
  • Golang迭代实现

简介

中序遍历(LDR)是二叉树遍历的一种,也叫做中根遍历、中序周游。

定义

在二叉树中,中序遍历首先遍历左子树,然后访问根结点,最后遍历右子树。若二叉树为空则结束返回,否则:(1)中序遍历左子树(2)访问根结点(3)中序遍历右子树
二叉树遍历——中序遍历(Golang)_第1张图片
如图所示二叉树,中序遍历结果:DBEAFC

Golang遍历实现

// TreeNode Definition for a binary tree node.
type TreeNode struct {
	Val   int       // 根
	Left  *TreeNode //左节点
	Right *TreeNode //右节点
}

func inorderTraversal(root *TreeNode) (res []int) {
	var inorder func(node *TreeNode)
	inorder = func(node *TreeNode) {
		if node == nil {
			return // 结束当前递归
		}
		inorder(node.Left)          // 直接怼到左边最下边
		res = append(res, node.Val) // 添加到数组里
		inorder(node.Right)         // 看右边还有没有分支,有就继续走,没有就将右节点加入数组
	}
	inorder(root)
	return
}

Golang迭代实现

栈:先进后出

// TreeNode Definition for a binary tree node.
type TreeNode struct {
	Val   int       // 根
	Left  *TreeNode //左节点
	Right *TreeNode //右节点
}

func inorderTraversal(root *TreeNode) (res []int) {
	stack := []*TreeNode{}    // 定义一个栈,栈存的就是一棵树
	for root != nil || len(stack) > 0 {
		for root != nil {
			stack = append(stack, root)  // 1.先将整颗树怼进去,在把所有的左子树怼进去
			root = root.Left   // 2.遍历左子树,直接左边的最下边
		}
		root = stack[len(stack)-1]  // 3.因为先进后出,拿到了最下面的左节点
		stack = stack[:len(stack)-1]
		res = append(res, root.Val) // 4.怼到数组里
                //5.看以右节点为根的还有没有左节点,有就回到上面第1步,没有就走第3步,把根节点
		root = root.Right // 
	}
	return
}

你可能感兴趣的:(笔记,Golang,深度优先,数据结构,算法)