[二叉数]二叉树的深度优先遍历（统一迭代法）

一、题目描述

原文链接：144. 二叉树的前序遍历


具体描述：
给你二叉树的根节点 root ，返回它节点值的 前序 遍历。

示例 1：

输入：root = [1,null,2,3]
输出：[1,2,3]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [1]
输出：[1]

示例 4：

输入：root = [1,2]
输出：[1,2]

示例 5：

输入：root = [1,null,2]
输出：[1,2]

提示：

• 树中节点数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

进阶：递归算法很简单，你可以通过迭代算法完成吗？

---

原文链接：145. 二叉树的后序遍历


具体描述：
给你一棵二叉树的根节点 root ，返回其节点值的 后序遍历 。

示例 1：

输入：root = [1,null,2,3]
输出：[3,2,1]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [1]
输出：[1]

提示：

• 树中节点的数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

进阶：递归算法很简单，你可以通过迭代算法完成吗？

---

原文链接：94. 二叉树的中序遍历


具体描述：
给定一个二叉树的根节点 root ，返回 它的 中序 遍历 。

示例 1：

输入：root = [1,null,2,3]
输出：[1,3,2]

示例 2：

输入：root = []
输出：[]

示例 3：

输入：root = [1]
输出：[1]

提示：

• 树中节点数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

二、思路分析

上次写的迭代法，前序和后序的写法差不多，但是中序个他们不一样，因为访问元素的顺序和中序的访问顺序是不一样的，所以不能写的一样！
但是看到递归的写法，如此的简单并且统一！羡慕呀，能不能把迭代的写法统一那，当然可以，而且并不难！

因为访问元素的顺序和深度优先遍历的顺序不一样，所以我们是不是可以先全部访问完，然后在开始处理元素那！
那还是用栈来做，根据栈的特点，在访问元素的时候要按照反顺序添加，如果前序是根左右，所以添加元素的时候是右左根！

三、AC代码

前序：

/**
 * 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> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();

        if (root != null) stack.push(root);

        while (stack.empty() == false){
            TreeNode node = stack.pop();
            if (node != null) {
                // 根左右==》右左根
                if (node.right != null) stack.push(node.right);
                if (node.left != null) stack.push(node.left);

                stack.push(node);
                stack.push(null);
                
            }else{
                result.add(stack.pop().val);
            }
        }

        return result;
    }
}

中序：

/**
 * 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> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        if (root != null) stack.push(root);
        while (stack.empty() == false){
            TreeNode node = stack.pop();
            if (node != null){
                // 先添加右，在根，在左
                if (node.right != null) stack.push(node.right);

                stack.push(node);
                stack.push(null);// 标记根节点，尚未处理

                if (node.left != null) stack.push(node.left);

            }else{// 当前元素为空，说明到了需要处理的时候
                result.add(stack.pop().val);
            }
        }

        return result;
    }

}

后序：

/**
 * 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) {
        List<Integer> result = new ArrayList<>();
        Stack<TreeNode> stack = new Stack<>();
        if (root != null) stack.push(root);
        while (stack.empty() == false){
            TreeNode node = stack.pop();
            
            if (node != null){
                // 根右左
                stack.push(node);
                stack.push(null);

                if (node.right != null) stack.push(node.right);

                if (node.left != null) stack.push(node.left); 
            }else {
                result.add(stack.pop().val);
            }
        }
        return result;
    }
}

四、总结

• 跟递归一样统一写法，还是很香的～

感谢大家的阅读，我是Alson_Code，一个喜欢把简单问题复杂化，把复杂问题简单化的程序猿！ ❤