解题思路分析
递归版本参考了邓俊辉老师的教材。在介绍具体方法之前,强调一下语义:访问一个节点,就是对这个节点进行操作,例如压入列表呀,用visit
来表示。而仅仅是遍历到一个节点,但是不做访问,仅仅做一些辅助动作,例如压栈呀,用go
来表示
下图是先序遍历的顺序图,可以看到对于先序遍历,都是首先沿着最左边一条路不停的向下,然后进行回溯向上。
可以将上图进一步抽象,先序遍历由2部分组成:
- 自顶而下的访问最左侧通路的节点
- 自底而上的访问最左侧通路上每一节点的右子树。其中每一个右子树也是先序遍历,也就是说每一个右子树也遵循这两部分
这里首先自顶而下,然后自底而上,我们可以用栈来记录这种关系。对于每一个二叉树(或者子树),首先沿着最左侧通路访问,在访问每一个节点的同时,将该节点的右孩子入栈(这样子自顶而下访问最左侧通路的同时,就自然的自底而上的访问其对应的右子树)
因此,二叉树的先序遍历可以总结为:
- 给定一个二叉树,首先沿着最左侧通路自顶而下访问(压入列表等操作),在访问的同时将每一个节点的右孩子压入栈中。根据栈的特性,自顶而上访问最左侧通路节点,就可以自底而上的访问对应的右子树
- 第一步走完之后,肯定是走到一个空节点,并且沿途的节点都访问完了,如果这个时候栈为空,说明刚刚访问过的节点都没有右子树,那不就说明所有节点都访问完了么?此时就可以结束操作,返回了
- 如果此时栈不为空,就可以将栈顶元素弹出,弹出之后我们要干什么?当然是以先序遍历的方式访问
以栈顶元素为根的二叉树
了,这样子就又回到第一步了,也就是将1,2,3步套在一个循环里面,在3步中,将栈顶元素弹出后赋值给第1步中的二叉树的根节点。
先序遍历以root为根的二叉树
定义一个node,首先node = root
while(true) {
1、以node为根的二叉树,沿着最左侧通路访问沿途节点,并将对应的右孩子压入栈中;
2、如果栈为空,说明全部节点都访问完了,break
3、如果栈不为空,就需要将栈顶元素topNode弹出, 将topNode赋值给node(以先序遍历的方式访问以topNode为根的子树)
}
代码实现
public class Solution03 {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<>();
Deque<TreeNode> stack = new ArrayDeque<>();
TreeNode node = root;
while (true) {
visitAloneLeft(node, stack, result);
if (stack.isEmpty()) {
break;
}
node = stack.removeLast();
}
return result;
}
// 以node为根的子树,沿着最左侧通路访问,将沿途节点的右孩子压入栈中
private void visitAloneLeft(TreeNode node, Deque<TreeNode> stack, List<Integer> result) {
while (node != null) {
result.add(node.val);
if (node.right != null) {
stack.addLast(node.right);
}
node = node.left;
}
}
}