leetcode刷题（34）——94.二叉树的中序遍历

一、题目

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

二、思路（递归——来自官方题解）

利用栈来实现。

创建一个 list 存储遍历结果。我们知道，中序遍历的顺序是：左子树------》根节点------》右子树，所以我们先把根节点入栈，然后从根节点开始，依次将其所有的左孩子入栈，如此，栈顶元素为左叶子节点。只要栈不为空，就弹出一个元素，将其存入 list 并把它的右子树（包括右孩子及其子树）入栈；遍历完整棵树并且 stack 为空时遍历完成。

举个例子，二叉树如图，遍历过程如下：

• 创建一个 list 存储遍历结果，以及一个辅助栈 stack；
• 从根节点开始，所有的左孩子依次入栈，得到 stack
  | 4 |
  | 2 |
  | 1 |
• 弹出栈顶元素 4，存入 list 得到 [4]，其右孩子为空，则继续弹栈；
• 弹出栈顶元素 2，存入 list 得到 [4，2]，将其右孩子（右孩子的子树为空）入栈，得到
  | 5 |
  | 1 |
• 弹出栈顶元素 5，存入 list 得到 [4，2，5]，其右孩子为空，则继续弹栈；
• 弹出栈顶元素 1，存入 list 得到 [4，2，5，1]，将其右孩子及右孩子的子树入栈，得到
  | 6 |
  | 3 |
• 弹出栈顶元素 6，存入 list 得到 [4，2，5，1，6]，其右孩子为空，则继续弹栈；
• 弹出栈顶元素 3，存入 list 得到 [4，2，5，1，6，3]，此时整棵树遍历完且栈为空，结束。

三、代码

class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        ArrayList<Integer> list = new ArrayList<>();
        LinkedList<TreeNode> stack = new LinkedList<>();
        TreeNode node = root;
        while(node != null || !stack.isEmpty()){
            while(node != null){
                stack.push(node);
                node = node.left;
            }
            node = stack.pop();
            list.add(node.val);
            node = node.right;
        }
        return list;
    }
}