[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<Integer> inorderTraversal(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
List<Integer> 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;
}
}
}
}