迭代法后序遍历二叉树

---

前言

二叉树后序遍历迭代法。

---

一、思路解析

1.整体思路

本文采用stack配合双指针法来迭代遍历二叉树。
双指针：cru指针负责将节点push到stack中，pre指针协助判断右子树是否已经遍历
难点理解：cru指针如何移动，何时给pre指针的赋值，如何判断右子树遍历情况

二、图解

1.初始赋值

2. 循环push左节点进栈

3. 左节点入栈完毕，右子树尚未遍历

4.调整cru和pre指针，准备遍历右子树

if(stack.peek().right == null || stack.peek().right == pre){
	pre = stack.pop();
	result.add(pre.val);
}else{
	/*
		此处代码为图示4即将运行的部分
		cru和pre指针指向当前未遍历的右子树
	*/
	cru = stack.peek().right;
	pre = cru;
}

5.右子树为空的特殊情况

特殊情况1.右子树为空，则无需遍历右子树，弹出stack中的当前节点（不是cru！！！）。
并且cru和pre指针都不用调整！！

/*
stack.peek().right == null:特殊情况右子树为空时
*/
if(stack.peek().right == null || stack.peek().right == pre){
	pre = stack.pop();
	result.add(pre.val);
}else{
	cru = stack.peek().right;
	pre = cru;
}

6.右子树已经遍历的特殊情况

特殊情况2.右子树已经遍历，则无需遍历右子树，弹出stack中的当前节点（不是cru！！！）。
并且cru和pre指针都不用调整！！

/*
  stack.peek().right == pre
  stack.peek() ==> 1号节点
	stack.peek().right ==> 2号节点
	pre ==> 2号节点
*/
if(stack.peek().right == null || stack.peek().right == pre){
	pre = stack.pop();
	result.add(pre.val);
}else{
	cru = stack.peek().right;
	pre = cru;
}

if(stack.peek().right == null || stack.peek().right == pre){
	pre = stack.pop();
	result.add(pre.val);
}else{
/*
  stack.peek().right ==> 4号节点
  pre ==> 2号节点
*/
	cru = stack.peek().right;
	pre = cru;
}

三、代码实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
    		//两根指针
        TreeNode cru = root;
        TreeNode pre = null;
        //读取的节点一次存入数组中
        ArrayList<Integer> result = new ArrayList<>();
        //迭代使用的栈
        Stack<TreeNode> stack = new Stack<>();
        
        while (!stack.empty() || cru != null) {
            if (cru != null) {
            //循环将左节点入栈
                stack.push(cru);
                cru = cru.left;
            } else {
            		//左节点遍历到尽头时，判断栈顶右子树情况：是否为空，是否已经遍历
                if(stack.peek().right == null || stack.peek().right == pre){
                		//右子树已经遍历：弹出栈顶节点，并用pre标记。
                		//同时准备查看下一个栈顶节点的右子树遍历情况
                    pre = stack.pop();
                    //存入被弹出节点
                    result.add(pre.val);
                }else{
                    //右子树还未遍历
                    cru = stack.peek().right;
                    pre = cru;
                }
            }
        }
        return result;
    }
}

四、难点总结

1.pre的理解：

1. 记录 栈顶节点的右子树遍历情况
2. 赋值时机：第1种情况：栈顶节点的右子树无需再入栈遍历时（右子树为空，或者已经遍历），pre = stack.pop（）。
   第2种情况：右子树需要遍历时，pre = cru = 右子树。
   总结可发现，只要左节点遍历到尽头时，pre指针总会开始移动到新的位置（已访问过的节点，但不一定是cru）。

2.cru的理解

1. 指谁，谁就入栈
2. 注意：cru为空表明左子树已经遍历完毕，此时如果右子树也已经遍历时，下一步要做的不是移动cru指针，而是循环判断栈顶节点的右子树是否已经遍历，直到找到右子树未被遍历的栈顶节点，cru和pre就指向该右子树

五、整体流程