GO语言遍历二叉树(非递归方式)

361 阅读1分钟

首先先序遍历二叉树,之前用到递归方式,先将本节点的数据先进行输出,再进行递归访问左子树和右子树,我们也可以使用栈的方式,将二叉树进行遍历

  • 首先思考什么是偶退出遍历循环,应该是节点为空或者栈为空,只有两者都符合条件,也就是,如果如果所有节点都遍历完成,或者放到栈中的数据全部被弹出,才遍历完成
  • 第二判断遍历的节点是否为空,如果不为空,则把当前节点的指针,放入栈,并把当前节点换为下一个节点的左树的节点。
  • 如果为空,从栈中提取出上一个节点,并将上一个节点进行弹出,将从栈中提取的上一个节点的右子树的节点赋值给进行循环的节点,
package main

type Student struct {
	name  string
	lnode *Student
	rnode *Student
}

func main() {
	student := Student{name: "xiang"}
	stu := Student{name: "left"}
	stu1 := Student{name: "right"}
	student.lnode = &stu
	student.rnode = &stu1
	node := &student
	stake := make([]*Student, 0)
	for node != nil || len(stake) != 0 {//当二叉树的节点不为空或栈中存储的数据不为空时
		if node != nil {//如果节点指针不为空
			stake = append(stake, node)//将此节点指针存入数组
			node = node.lnode//将此节点换为此节点的左子树
		} else {
			node = stake[len(stake)-1]//如果节点指针为空,则弹出上一个节点的指针
			stake = stake[:len(stake)-1]//将上一个节点的指针进行出栈操作
			node = node.rnode//将此节点换为此节点的右子树
		}
	}
}

阅读世界,共赴山海

423全民读书节,邀你共读