给定一个二叉树,返回它的 后序 遍历。
示例:
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
递归
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
walk(root,res);
return res;
}
private void walk(TreeNode root, List<Integer> res) {
if(null == root) {
return ;
}
walk(root.left, res);
walk(root.right, res);
res.add(root.val);
}
}
迭代
后序遍历的顺序是左 右 根, 前序遍历的顺序是 根 左 右 ,基于 前序遍历 做下修改,使其遍历顺序为 根 右 左,然后将结果集进行反转.
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
while (null != root || ! stack.isEmpty()) {
while (null != root) {
res.add(root.val);
stack.add(root.left);
root = root.right;
}
root = stack.pop();
}
Collections.reverse(res);
return res;
}
}