遍历一棵二叉树最简单的方法就是采用递归法,也就是DFS,比较简单。
前序遍历:
public void dfs(TreeNode root) {
visit(root);
dfs(root.left);
dfs(root.right);
}
中序遍历:
public void dfs(TreeNode root) {
dfs(root.left);
visit(root);
dfs(root.right);
}
后序遍历:
public void dfs(TreeNode root) {
dfs(root.left);
dfs(root.right);
visit(root);
}
对于dfs,这里不再解释,我们主要来看一下如何采用非递归的方式来遍历二叉树*(੭ˊᵕˋ)੭ଘ
前序遍历
思路:**我们采用栈的方式来遍历二叉树,每次先加入到结果集中的是根节点,然后右节点和左节点入栈,为何要右节点先入栈呢?因为先入栈的节点会后出来,那么这样左节点才能先于右节点被遍历。**完整代码如下:
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> ret = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pop();
if (node == null) continue;
ret.add(node.val);
stack.push