题目
给你一棵二叉树的根节点 root ,返回其节点值的 后序遍历 。
示例 1:
输入:root = [1,null,2,3]
输出:[3,2,1]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
提示:
树中节点的数目在范围 [0, 100] 内
-100 <= Node.val <= 100
进阶:递归算法很简单,你可以通过迭代算法完成吗?
思路1:递归
代码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 {
List<Integer> ret = new LinkedList<>();
public List<Integer> postorderTraversal(TreeNode root) {
if(root == null) {
return ret;
}
postorderTraversal(root.left);
postorderTraversal(root.right);
ret.add(root.val);
return ret;
}
}
思路2:
代码2
/**
* 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 {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> ret = new LinkedList<>();
if(root == null) {
return ret;
}
Stack<TreeNode> stack = new Stack<>();
//cur表示当前需要判断的节点
TreeNode cur = root;
//prev表示最近一次完全结束遍历的节点
TreeNode prev = null;
while(cur != null || !(stack.isEmpty())) {
//一路向左走到头
while(cur != null) {
stack.push(cur);
cur = cur.left;
}
//左树已经全部结束
cur = stack.pop();
//此时cur是第二次访问,不能直接遍历,需要判断右树是否为空或者是否遍历结束
if(cur.right == null || prev == cur.right) {
//右树已经结束遍历,第三次访问cur
ret.add(cur.val);
//此时cur已经全部访问完毕,prev更新为当前cur节点
prev = cur;
cur = null;
} else {
stack.push(cur);
//继续访问右子树
cur = cur.right;
}
}
return ret;
}
}
前、中、后序遍历借助栈实现,根节点啥时候能真正遍历输出?
- 前:第一次访问根节点就可以输出。
- 中:第二次访问根节点(走完左树)可以输出。
- 后:第三次访问根节点(走完子树)可以输出。
节点入栈和出栈:
- 入栈:第一次走到节点入栈。
- 出栈:当真正访问完这个节点出栈。(左、右、根)