【复习笔记】二叉树的前中后非递归遍历算法

前序遍历

思路：全局指针，指向根节点，所到节点立即访问，然后把节点入栈，往左走。

当左子树为空时，pop栈顶节点，如果栈顶右子树不空，把全局指针往右走，继续循环。

循环结束条件：栈为空并且全局指针为null时。

public static void preIte(TreeNode root){Stack<TreeNode> stack = new Stack<TreeNode>();TreeNode temp = root;while(!stack.isEmpty() || temp != null){if(temp != null){System.out.println(temp.val);stack.add(temp);temp = temp.left;}else{temp = stack.pop();temp = temp.right;}}}

中序遍历

思路：全局指针，初始指向根节点，循环入栈，直到全局指针的左子树为空。

pop栈顶节点并访问，如果栈顶节点的右子树不为空，则全局指针往右走，继续循环。

循环结束条件：栈空或者全局指针为null时。

public static void midIte(TreeNode root){Stack<TreeNode> stack = new Stack<TreeNode>();TreeNode temp = root;while(temp != null || !stack.isEmpty()){while(temp != null){stack.add(temp);temp = temp.left;}if(!stack.isEmpty()){temp = stack.pop();System.out.println(temp.val);temp = temp.right;}}}

后序遍历

思路：全局指针+上一次遍历节点的指针，首先根节点入栈。

如果全局指针现在指向的节点没有左右孩子 或者 全局指针指向的节点的左孩子或者右孩子是上一次遍历的节点，则

遍历当前节点，并出栈，同时上一次遍历的节点指向此节点。否则，当前节点的右孩子若不为空，右孩子入栈，当前节点的左孩子不为空，左孩子入栈。

循环结束条件：栈空。

public static void postIte(TreeNode root){Stack<TreeNode> stack = new Stack<TreeNode>();TreeNode temp = root;TreeNode preNode = null;if(temp != null) stack.add(temp);while(!stack.isEmpty()){temp = stack.peek();if((temp.right == null && temp.left == null) ||(preNode != null && (temp.left == preNode || temp.right == preNode))){System.out.println(temp.val);stack.pop();preNode = temp;}else{if(temp.right != null)stack.add(temp.right);if(temp.left != null)stack.add(temp.left);}}}