给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [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() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
Deque<TreeNode> stack = new LinkedList<>();
LinkedList<Integer> list = new LinkedList<>();
public List<Integer> postorderTraversal(TreeNode root) {
lrd(root);
return list;
}
private void lrd(TreeNode root){
if(root == null) return;
stack.addFirst(root);
while(root != null && !stack.isEmpty()){
TreeNode node = stack.removeFirst();
list.addFirst(node.val);
if(node.left != null){stack.addFirst(node.left);}
if(node.right != null){stack.addFirst(node.right);}
}
}
}