二叉树的遍历,也可以使用非递归的方法实现,下面就介绍一下它的代码实现。
首先,先要完成节点类的定义:
public class TreeNode {
int value;
TreeNode left; //左孩子
TreeNode right; //右孩子
public TreeNode() {
value = 0;
left = right = null;
}
public TreeNode(int value) {
this.value = value;
left = right = null;
}
}
前序遍历、中序遍历、后序遍历都可以利用栈,来实现遍历,这些遍历方法,都比较容易理解,在代码实现上来说,递归的方式更加让人理解,对于非递归来说,相对不是很容更易理解,但是便利的过程是一样的,在非递归的方法中也是有需要注意的地方的。
1、前序遍历
//前序遍历
public void preOrder(TreeNode root) {
Stack<TreeNode> st = new Stack<>();
TreeNode curNode = root;
while (curNode != null || !st.isEmpty()) {
while (curNode != null) {
System.out.print(curNode.value+" "); //首先访问根
st.push(curNode); //访问根的左子树
curNode = curNode.left;
}
//获取栈顶元素
curNode = st.pop();
//访问栈顶元素的右子树
curNode = curNode.right;
}
}
2、中序遍历
中序遍历:与前序遍历相似,注意是左根右。
//中序遍历
public void inOrder(TreeNode root) {
Stack<TreeNode> st = new Stack<>();
TreeNode curNode = root;
while (curNode != null || !st.isEmpty()) {
while (curNode != null) {
st.push(curNode);
curNode = curNode.left;
}
curNode = st.pop();
System.out.print(curNode.value+" ");
curNode = curNode.right;
}
}
3、后序遍历
后序遍历更加要注意,不能扰乱顺序,应先访问右孩子,不能像上述的两种遍历方法去压栈,要注意先判断右孩子是否访问到。这点是非常需要注意的。注意下方代码的注释部分:
//后序遍历
public void postOrder(TreeNode root) {
Stack<TreeNode> st = new Stack<>();
TreeNode curNode = root;
TreeNode prev = null;
while (curNode != null || !st.isEmpty()) {
while (curNode != null) {
st.push(curNode);
curNode = curNode.left;
}
//根: 访问完右子树之后,才能访问根
//curNode = st.pop(); 有可能导致元素访问不到
curNode = st.peek();
if (curNode.right == null || curNode.right == prev) {
System.out.print(curNode.value+" ");
prev = curNode;
st.pop();
curNode = null;
} else {
curNode = curNode.right;
}
}
}
以上是二叉树的三种遍历方法的非递归实现。