二叉树的非递归遍历遵循以下套路:
1.确定二叉树的遍历方式,根据方式确定curr的维护规律,此代表的是节点入栈的规律;
2.寻找节点出栈的时机,节点出栈代表要处理节点;
3.出栈一次节点,维护下一次的curr
用此模式分析二叉树的前序遍历:
前序遍历的顺序是先处理当前节点,再处理左节点和右节点。按照这种规律,必定要在使用栈来维护遍历顺序时,先将当前节点从栈内弹出,之后寻找下一个入栈节点,即下一步curr的维护。
所以,前序遍历的思路为:对当前curr入栈后,即弹栈去操作,之后,将右节点加入栈,再将左节点入栈,curr代表先右后左。
循环逻辑:判断栈是否有节点,有的话弹出栈顶节点,处理完毕分别将右节点和左节点入栈(如果有的话)
用此模式分析二叉树的中序遍历:
中序遍历对节点的处理顺序是左中右,所以左侧节点要先于中间节点出栈,遍历过程中,依照左侧优先的顺序入栈curr节点,当curr为空,表示需要处理栈顶节点,此时pop栈顶节点,处理后访问右节点,使用curr表示。
循环条件:栈不为空或者curr所表示的节点不为空。
public static void inorderNR(Main.Node node) {
Stack<Main.Node> stack = new Stack<>();
Main.Node curr = node;
//curr表示当前访问到的节点
while (!stack.isEmpty() || curr != null) {
//如果当前寻访到的节点不为空
if (curr != null) {
//就先把它入栈,待之后出栈处理
stack.push(curr);
//朝左子树方向寻找它的叶子节点
curr = curr.left;
} else {
//curr为空说明到了[某个子树]的叶子节点
//不再需要寻访了,可以出栈去处理了
Main.Node pop = stack.pop();
System.out.println(pop);
//处理完当前的之后,按照左中右的顺序,紧接着去寻访右子节点
if (pop.right != null) {
curr = pop.right;
}
}
}
}
用此模式分析二叉树的后序遍历:
后序遍历对节点的处理顺序是左右中,也是依照左侧优先的顺序入栈curr节点,左节点为空则寻找右节点,当curr为空,则将栈顶节点弹出,处理后判断当前节点若为栈顶左节点,则需要继续寻访栈顶节点的右节点,否则,curr就置为空,为下一步循环,pop栈顶操作做准备。
public static void postorderNR(Main.Node root) {
Stack<Main.Node> stack = new Stack<>();
//curr表示每次循环之下可以访问到的节点
Main.Node curr = root;
//循环条件为栈不为空或者当前需要访问的节点不为空
while (!stack.isEmpty() || curr != null) {
//每个当前在访问的节点都要不停寻访到它子树的最左侧叶子节点
while (curr != null) {
//将寻访路径的节点依次入栈
stack.push(curr);
//不停寻访子节点,左子树优先
if (curr.left != null) {
curr = curr.left;
} else {
curr = curr.right;
}
}
//寻访结束时curr为空,表示到了叶子节点了,开始自底向上出栈了
Main.Node pop = stack.pop();
System.out.println(pop); //处理出栈的节点
//处理完这个节点就要考虑它是否为它父节点(此时栈顶的节点)的左孩子
//如果是的话,按照[后序遍历-左右中]的处理顺序,需要去寻访父节点的右孩子了
if (!stack.isEmpty() && stack.peek().left == pop) {
curr = stack.peek().right;
}
}
}