[LeetCode] 94. 中序遍历二叉树

传送门

题目描述

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

例如：
给定二叉树 [1,null,2,3],

   1
    \
     2
    /
   3

返回 [1,3,2].

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

我的想法

题中说明了要求使用迭代法

使用一个栈来存储二叉树节点，根据中序遍历的规则，我们可以推算出这样的规律：
1. 将当前非空节点入栈
2. 如果左子节点不为空，则继续将左子节点入栈
3. 如果左子节点为空，则抛出栈顶节点并记录 val 值，然后将其右子节点入栈
4. 重复 1、2、3 步骤直至栈空

AC 代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List inorderTraversal(TreeNode root) {
        Stack stack = new Stack<>();
        List list = new ArrayList<>();
        while(true){
            if (root != null){
                stack.push(root);
                root = root.left;
            } else {
                if (stack.empty()) {
                    return list;
                }
                root = stack.pop();
                list.add(root.val);
                root = root.right;
            }
        }
    }
}