根据中序遍历的顺序,对于任一结点,优先访问其左孩子,而左孩子结点又可以看做一根结点,然后继续访问其左孩子结点,直到遇到左孩子结点为空的结点才进行访问,然后按相同的规则访问其右子树。因此其处理过程如下:
对于任一结点root,引入一个辅助节点p,其作用是:标记已经访问过的节点,
1)将root压入栈中,只有有左孩子,就压入栈中
if(p!=null && p.left!=null) {
stk.add(p.left);
p = p.left;
}
2)对栈顶元素 q 进行出栈操作,访问该栈顶结点。
如果 q 没有右孩子,将辅助节点 p 设置为null。表示下一步还是进行2)操作,出栈操作
如果 q 有右孩子,将右孩子压栈中,p = q的右孩子
p = stk.pop();//弹出栈顶节点 左孩子--->根节点
System.out.print(p.val+" ");//访问
if(p!=null && p.right!=null) {//如果栈点元素有右孩子的话,将有节点压入栈中
stk.add(p.right);
p = p.right;
}else
p = null;//p=stk.pop;已经访问过p了,p设置为null
3)直到栈为空,遍历结束。
代码重点是,2先访问出栈,然后将5节点呀入栈中
import java.util.Stack;
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Main {
//中序遍历 递归算法
public static void InOrder(TreeNode root) {
if(root==null)return;
InOrder(root.left);
System.out.print(root.val+" ");
InOrder(root.right);
}
// 中序遍历 非递归算法
public static void InOrder2(TreeNode root) {
if(root==null)return;
Stack<TreeNode> stk = new Stack<TreeNode>();
TreeNode p = root;//辅助节点
stk.add(p);
while(stk.isEmpty() == false) {
//只要你有左孩子,就将左孩子压入栈中
if(p!=null && p.left!=null) {
stk.add(p.left);
p = p.left;
}else {
p = stk.pop();//弹出栈顶节点 左孩子--->根节点
System.out.print(p.val+" ");//访问
if(p!=null && p.right!=null) {//如果栈点元素有右孩子的话,将有节点压入栈中
stk.add(p.right);
p = p.right;
}else
p = null;//p=stk.pop;已经访问过p了,p设置为null
}
}
}
public static void main(String[] args) {
TreeNode root = new TreeNode(1);
root.left = new TreeNode(2);
root.right = new TreeNode(3);
root.left.left = new TreeNode(4);
root.left.right = new TreeNode(5);
root.right.left = new TreeNode(6);
root.right.right = new TreeNode(7);
InOrder2(root);
}
}