解题思路分析
参考了邓俊辉老师的教材。
中序遍历:访问左子树、访问当前节点、访问右子树。
中序遍历的顺序如下图所示
可以发现对于每一个子树的中序遍历,总是先找到最左侧通路最下面的节点。然后访问它的右孩子,最后向上回溯。因此将中序遍历的顺序进一步抽象,如下图所示:
如上图所示,图中最左侧通路一共有L0 L1 ....Ld
,d+1
个节点,因此中序遍历可以分为d+1
端。每一段都可以由最左侧通路上的节点和其对应的右子树组成。当我们访问完一个节点和他对应的右子树后,我们需要回溯到最左侧通路上的上一个节点。例如访问完了Ld
,就访问对应的右子树Td
,这一段访问完之后,需要回溯到Ld-1
,因此需要用栈来帮助这种回溯。
中序遍历可以总结为:
- 对于以root为根的二叉树,首先沿着最左侧通路行走(go,不做访问),并将沿途的节点压入栈中
- 第1步完成之后,节点肯定为空,此时如果栈为空,说明这个二叉树里面没有节点,那就可以返回了
- 如果栈不为空,就需要将栈顶元素topNode弹出,这个栈顶元素就是刚刚走过的
最左侧通路的最下面的节点
,根据上面的分析,我们需要访问这个节点,并以中序遍历的方式访问它的右子树,就又回到了第1步,也就是将1,2,3步套在一个循环里,将topNode赋值给1中的root
以中序遍历的方式访问以root为根的二叉树
定义一个node,初始化为root
while (true) {
1、以node为根的二叉树,沿着最左侧通路行走,并将沿途的节点压入栈中;
2、如果栈为空,break;
3、如果栈不为空,弹出栈顶元素topNode,访问栈顶元素,将node = topNode.right
}
代码实现
// 中序遍历,迭代版本
public class Solution02 {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<>();
Deque<TreeNode> stack = new ArrayDeque<>();
TreeNode node = root;
while (true) {
goAloneLeft(node, stack);
if (stack.isEmpty()) {
break;
}
TreeNode topNode = stack.removeLast();
result.add(topNode.val);
node = topNode.right;
}
return result;
}
// 以node为根的子树,沿着最侧通路行走,将沿途的节点入栈,直到节点为空
private void goAloneLeft(TreeNode node, Deque<TreeNode> stack) {
while(node != null) {
stack.addLast(node);
node = node.left;
}
}
}