145. 二叉树的后序遍历 递归解法与非递归解法
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
解法1:递归
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> list = new LinkedList<>();
postorderTraversal(root,list);
return list;
}
public void postorderTraversal(TreeNode root, List<Integer> list) {
if (root==null){
return;
}
postorderTraversal(root.left,list);
postorderTraversal(root.right,list);
list.add(root.val);
}
解法二:迭代
模板
二叉树的遍历都可以基于这个模板想出来,在这个基础之上仔细思考思考总能想出来
while( 栈非空 || p 非空)
{
if( p 非空)
{
}
else
{
}
}
解题思路
1.设置node=root
2.如果node是叶子节点或则该节点的左右节点都被访问过了则访问该元素并出栈
否则将该节点的右节点和左节点依次入栈
public List<Integer> postorderTraversal(TreeNode root) {
LinkedList<Integer> list = new LinkedList<>();
if (root==null){
return list ;
}
Stack<TreeNode> stack = new Stack<>();
TreeNode node=root;
TreeNode pre=null;
while (!stack.isEmpty()||node!=null){
if (node!=null){
stack.add(node);
node=node.left;
}else{
node = stack.peek();
/*如果当前节点的右节点为空获取右节点已经被访问过了*/
if(node.right == null || node.right == pre)
{
list.add(node.val);
pre = node;
stack.pop();
/*将node置为null取出下一个栈顶节点*/
node = null;
}
else{
node = node.right;
}
}
}
return list;
}