Leetcode 94. 二叉树的中序遍历 Binary Tree Inorder Traversal (Java)

Leetcode 94. 二叉树的中序遍历Binary Tree Inorder Traversal (Java)

##Hash Table##, ##Stack##, ##Tree##

题解

二叉树的中序遍历

递归

按照左子树=>根节点=>右子树的顺序遍历

时间复杂度：O(n)

每个结点访问一次

空间复杂度为层数

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        ArrayList<Integer> res = new ArrayList<Integer>();
        traversal(root, res);
        return res;
    }
    
    public void traversal(TreeNode root, ArrayList<Integer> path) {
        if (root == null) return;
        traversal(root.left, path);
        path.add(root.val);
        traversal(root.right, path);
    }
}

---

迭代

借助栈模拟递归实现迭代法

• 循环迭代树结点
• 树结点有左子树时，遍历该节点但不读出，该节点进栈，该过程可递归操作
• 直到最后一个没有左子树的结点
• 弹出栈顶元素，并读出该结点
• 用与遍历左子树相同的方式遍历该节点的右子树
• 栈为空循环结束

**时间复杂度：**O(n)

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new LinkedList<Integer>();
        Stack<TreeNode> sta = new Stack<>();
        TreeNode cur = root;
        
        while (cur != null || !sta.isEmpty()) {
            if (cur != null) {
                sta.push(cur);
                cur = cur.left;
            } else {
                cur = sta.pop();
                res.add(cur.val);
                cur = cur.right;
            }
        }
        return res;
    }
}

---

通用迭代法（使用该方法能够统一前中后序遍历）

Reference: https://www.acwing.com/solution/content/176/
https://mp.weixin.qq.com/s/WKg0Ty1_3SZkztpHubZPRg
二叉树前序遍历(通用迭代法)
二叉树后序遍历(通用迭代法)

需要解决的问题是访问结点（遍历结点）和处理结点（将元素放入结果集）的处理时间不一致的问题

即访问到某结点时，并不会立即加入结果集，而是需要先遍历完其左子树在加入结果集的问题

因此，可以将访问的结点都放入栈中，但是将需要处理的结点做标记

标记的方法为在需要处理的结点放入栈后，紧接着放一个空指针作为标记

入栈顺序为右子树=>中间结点=>左子树

时间复杂度： O(n)

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        Stack<TreeNode> sta = new Stack<TreeNode>();
        if (root != null) sta.push(root);
        TreeNode curr;
        
        while (!sta.isEmpty()) {
            curr = sta.pop(); //先取出栈顶元素，避免后续重复操作，后续操作中可能会重新入栈
            if (curr != null) {
                if (curr.right != null) sta.push(curr.right); //右子树为空时，不放入空结点，避免出现栈中有两个相邻的空结点
                sta.push(curr);  // 放入左节点
                sta.push(null);  // 放入左节点后放入空结点作为标记
                if (curr.left != null) sta.push(curr.left); // 左子树为空时，不放入空结点
            } else { //栈顶元素为空时，代表需要处理元素
                curr = sta.pop();  
                res.add(curr.val);
            }
        }
        return res;
    }
}