ps:看完这篇文章你将掌握二叉树的非递归前序遍历、中序遍历和后续遍历,以及它们在leetcode上一些题如#94,#538,#938等题的应用
一、前序遍历
首先说前序遍历,要了解前序遍历的非递归写法,首先要弄明白前序遍历的遍历策略。考虑一种最简单的二叉树模型,模型中包括了根节点,左子树和右子树,在前序遍历中,对于这样一颗二叉树,前序遍历的意思就是一定是以Root节点、左子树、右子树的顺序一次遍历各个节点,当遍历左子树时左子树又可以看作另一颗二叉树,因此这是一种典型递归的思想。
注意到上述遍历过程中,遍历左子树过后需要先返回到根节点,再通过根节点遍历右子树,因此非递归需要通过栈来实现,其实递归在JVM虚拟机中也是通过栈表现出来的。那么现在来缕一下思绪,前序节点的遍历思路如下
- 如果当前节点不为空,就先输出,并将自身保存到栈中
- 如果当前节点为空,那么回退一步通过栈找到上一节点
- 将上一节点的右子节点作为当前节点,重新开始循环
对应代码如下:
public static void traverse(TreeNode root){
//实现前序遍历
Deque<TreeNode> stack = new ArrayDeque<>();
while(!(stack.isEmpty()&&root==null)){
if(root!=null){
stack.addLast(root);
System.out.print(root.val+",");//前序
root = root.left;
}else{
root = stack.pollLast();
root = root.right;
}
}
}
二、中序遍历
中序遍历其实和前序遍历非常相似,其遍历思路整体和前序是一摸一样的,只是输出位置有所变化,中序遍历思路如下:
- 如果当前节点不为空,则将自身保存到栈中
- 如果当前节点为空,那么回退一步通过栈找到上一节点并输出上一节点
- 将上一节点的右子节点作为当前节点,重新开始循环
对应代码如下:
public static void traverse(TreeNode root){
//实现中序遍历
Deque<TreeNode> stack = new ArrayDeque