递归能做的迭代一定能做,只不过递归会复杂一些
迭代法实现前序遍历
题目:给你二叉树的根节点 root
,返回它节点值的 前序 遍历。
图示:
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
if (root == null) {
return res;
}
Deque<TreeNode> stack = new LinkedList<>();
TreeNode node = root;
while (!stack.isEmpty() || node != null) {
while (node != null) {
res.add(node.val);
stack.push(node);
node = node.left;
}
node = stack.pop();
node = node.right;
}
return res;
}
}
迭代法实现中序遍历
把 res 添加的位置调一下就可以了,要是看不懂的话应该是上面的没理解。
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
Deque<TreeNode> stack = new LinkedList<TreeNode>();
while (root != null || !stack.isEmpty()) {
while (root != null) {
stack.push(root);
root = root.left;
}
root = stack.pop();
res.add(root.val);
root = root.right;
}
return res;
}
迭代法实现后序遍历
总的来所右这么几种方法
- 反转法
- 访问标记法
- Morris法: 链接:https://zhuanlan.zhihu.com/p/101321696
我们重点讲 反转方
我们观察到,后续排序反转之后就变成了类似于前序排序的算法
public List<Integer> postOrderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) return res;
Stack<TreeNode> stack = new Stack<>();
TreeNode node = root;
while (!stack.isEmpty() || node != null) {
while (node != null) {
res.add(node.val);
stack.push(node);
node = node.right;
}
node = stack.pop();
node = node.left;
}
// 反转一下
Collections.reverse(res);
return res;
}