题目
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
递归
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
List<Integer> ans;
public List<Integer> postorderTraversal(TreeNode root) {
ans = new ArrayList<>();
dfs(root);
return ans;
}
public void dfs(TreeNode root) {
if (root == null) return ;
dfs(root.left);
dfs(root.right);
ans.add(root.val);
}
}
迭代1
自己写的,主要用了一个HashSet
记录访问过的节点。遍历的时候只有当右孩子为空或者访问过的时候才出栈并访问该节点,同时将其标记为以访问。为什么强调自己写的呢?因为官方题解还是比我的优雅!
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> ans = new ArrayList<>();
LinkedList<TreeNode> stack = new LinkedList<>();
Set<TreeNode> set = new HashSet<>();
if (root == null) return ans;
// 将空值放入set,这样判断空值的时候可以直接用set判断
set.add(null);
TreeNode p = root;
while (p != null) {
stack.add(p);
p = p.left;
}
while (!stack.isEmpty()) {
p = stack.getLast();
// 只有当前节点的右孩子为空或者以访问时,才访问出栈,同时用set记录
if (set.contains(p.right)) {
stack.pollLast();
ans.add(p.val);
set.add(p);
} else {
// 否则将其右孩子及右孩子最左边节点入栈
p = p.right;
while (p != null) {
stack.add(p);
p = p.left;
}
}
}
return ans;
}
}
迭代2
官方题解其实就是前序遍历,只是利用了LinkedList
链表的特性,每访问一个节点使用头插法插入链表,这样输出的时候就变成后序了!
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
LinkedList<TreeNode> stack = new LinkedList<>();
LinkedList<Integer> output = new LinkedList<>();
if (root == null) {
return output;
}
stack.add(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pollLast();
output.addFirst(node.val);
if (node.left != null) {
stack.add(node.left);
}
if (node.right != null) {
stack.add(node.right);
}
}
return output;
}
}