思路:
之前我们学习了使用递归法来进行二叉树的三种遍历,这次我们考虑使用迭代法来进行遍历,使用栈来实现。推荐将两篇参考文章都看一遍。在两篇参考文章中,第一篇讲了如何使用迭代法来遍历,但是三种遍历方式的实现并不是特别统一,没法像递归法一样仅修改中间元素的访问时机来非常简单地从一种遍历顺序改造为另一种遍历顺序,而是需要较大幅度修改代码。第二篇则讲了如何将这种迭代法改造为可以统一实现的方式。
本篇文章中所附代码使用的是第二篇参考文章中提及的统一实现的方式。
注意点:
1、推入栈的顺序需要注意,不论是哪种遍历顺序,左儿子总是比右儿子先遍历,但是在将他们推入栈的时候,注意是先将右儿子推入栈中,再把左儿子推入栈中,这样才能在把它们弹出栈进行访问时才是顺序正确的,在参考文章1中有细讲这里。
2、关于在栈中推入null节点,这是起到标记的作用,每当遇到null节点时,意味着我们已经将null节点的下一个节点的左右儿子推入到栈过了,这次我们可以直接将下一个节点的值输出了,在参考文章2中有细讲这里。
Java代码:
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if (root == null) return ans;
Deque<TreeNode> deque = new LinkedList<>();
deque.push(root);
while (!deque.isEmpty()) {
TreeNode rnNode = deque.pop();
if (rnNode == null) {
rnNode = deque.pop();
ans.add(rnNode.val);
} else {
if (rnNode.right != null) deque.push(rnNode.right);
if (rnNode.left != null) deque.push(rnNode.left);
deque.push(rnNode);
deque.push(null);
}
}
return ans;
}
}
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if (root == null) return ans;
Deque<TreeNode> deque = new LinkedList<>();
deque.push(root);
while (!deque.isEmpty()) {
TreeNode rnNode = deque.pop();
if (rnNode == null) {
rnNode = deque.pop();
ans.add(rnNode.val);
} else {
deque.push(rnNode);
deque.push(null);
if (rnNode.right != null) deque.push(rnNode.right);
if (rnNode.left != null) deque.push(rnNode.left);
}
}
return ans;
}
}
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> ans = new ArrayList<>();
if (root == null) return ans;
Deque<TreeNode> deque = new LinkedList<>();
deque.push(root);
while (!deque.isEmpty()) {
TreeNode rnNode = deque.pop();
if (rnNode == null) {
rnNode = deque.pop();
ans.add(rnNode.val);
} else {
if (rnNode.right != null) deque.push(rnNode.right);
deque.push(rnNode);
deque.push(null);
if (rnNode.left != null) deque.push(rnNode.left);
}
}
return ans;
}
}