LC 94.二叉树的中序遍历

二叉树的中序遍历

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

示例 1：

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

示例 2：

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

示例 3：

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

提示：

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

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

解法一(递归)

思路分析：

1. 根据二叉树的定义，使用递归来进行求解，对于如何实现递归需要思考以下三点

2. 递归函数的参数和返回值；对于题目要求获取节点值并保存到一个列表中，所以递归函数的参数有两个，同时也可以发现不需要有返回值

3. 递归的终止条件；因为采用深度优先搜索，所以会一直往下遍历，直到二叉树节点为空，则停止，所以终止条件为；节点为空

4. 单层递归的过程；由于采用中序遍历，所以需要先继续递归遍历左子节点，再读取当前节点的值，最后遍历右子节点

实现代码如下：

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

提交结果如下：

解答成功:
执行耗时:0 ms,击败了100.00% 的Java用户
内存消耗:40.4 MB,击败了5.01% 的Java用户

复杂度分析：

• 时间复杂度：$O(n)$，遍历一遍二叉树

• 空间复杂度：$O(n)$，递归调用

解法二(迭代)

思路分析：

1. 根据前序遍历迭代法，对于中序遍历顺序，左中右，首先将根节点所有左节点全部压入栈，遍历完左节点后，即开始遍历

2. 此时遍历左节点，遍历结束后，需要将左节点的父节点的左孩子设为null，避免重复遍历

3. 然后再对此时节点的右节点进行判断，若不为空，则压入栈

4. 若有当前节点的右节点压入栈中，则再下一轮迭代时，又会一直深入遍历右节点的左分支，直到左压入栈结束为止

实现代码如下：

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        if (root == null)    // 边界条件
            return ans;
        Deque<TreeNode> stack = new LinkedList<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            while (stack.peek().left != null)    // 一直往左 直到第一个应该遍历的左节点
                stack.push(stack.peek().left);
            TreeNode node = stack.pop();    // 弹出应该遍历的节点
            if(!stack.isEmpty())    // 切点该节点 与栈中可能是父节点的联系
                stack.peek().left = null;
            ans.add(node.val);    // 遍历
            if (node.right != null) // 若存在右节点 则开始遍历
                stack.push(node.right);
        }
        return ans;
    }
}

提交代码如下：

解答成功:
执行耗时:0 ms,击败了100.00% 的Java用户
内存消耗:40.1 MB,击败了11.80% 的Java用户

复杂度分析：

• 时间复杂度：$O(n)$

• 空间复杂度：$O(n)$

解法三(统一迭代法)

思路分析：

1. 将待访问节点和待处理节点放入栈中

2. 同时使用空指针来标记待处理节点

3. 即将栈中元素按照中序遍历顺序输出

实现代码如下：

class Solution {
    // 统一前中后序遍历的迭代法 中序
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> ans = new ArrayList<>();
        if (root == null)
            return ans;        // 处理边界条件
        Deque<TreeNode> stack = new LinkedList<>();
        stack.push(root);    // 先保存根节点
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            if (node != null) {
                // node = stack.pop();    // 将节点移除 避免重复操作
                // 先将右节点保存  因为右节点最后遍历
                if (node.right != null)
                    stack.push(node.right);
                // 再保存中节点
                stack.push(node);
                stack.push(null);    // 通过空值标记待处理节点
                // 最后保存左节点
                if (node.left != null)
                    stack.push(node.left);
            } else {
                // stack.pop();    // 将空结点移除 下一个节点才是需要访问的节点
                node = stack.pop();
                ans.add(node.val);
            }
        }
        return ans;
    }
}

提交代码如下：

解答成功:
执行耗时:0 ms,击败了100.00% 的Java用户
内存消耗:40.6 MB,击败了5.09% 的Java用户

复杂度分析：

• 时间复杂度分析：$O(n)$

• 空间复杂度分析：$O(n)$