难度:中等
频次:31
题目:给你一棵二叉树的根节点 root ,返回其节点值的 后序遍历 。
解题思路:
- 递归法
- 迭代法
递归法(DFS)
代码:
/**
* 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> res=new ArrayList<Integer>();
dfs(res,root);
return res;
}
//DFS 深度搜索遍历
public void dfs(List<Integer> res,TreeNode root){
//递归结束条件
if(root==null) return;
dfs(res,root.left);
dfs(res,root.right);
res.add(root.val);
}
}
迭代法(用栈模拟递归)
注意:
- 前中序遍历,中间的节点只需要遍历一次,但是后序遍历,需要遍历2次,并且在第二次时才需要记录
- 第一次和第二次的区别,就是右节点是否已经遍历过了,所以用一个指针pre标记已经遍历过的节点即可
- 当指针pre指向右节点,即是第二次中间节点了
- 遍历的条件时栈不为空(.size()>0)或者遍历的节点不为空===>即终止的条件就是root==null,并且栈里没有节点
- 遍历节点,如果节点不为空,把节点push到栈里,然后再遍历该节点的左节点
- 遍历节点,如果节点为空,把栈里的栈顶节点拿出来判断===》判断该节点的右节点是否已经遍历过1次了===》即判断右子树是否为空,或者右子树根节点已经遍历过一次了,每次遍历右节点的时候都需要
- 这里还需要注意一个点,就是这里pop出来的节点的右节点已经遍历,所以不能再遍历该节点的右节点,所以这里必须要有一个新的top用来指向
代码
/**
* 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> res=new ArrayList<>();
Stack<TreeNode> stack=new Stack<>();
//用来记录已经遍历的节点==因为第二次遍历中间节点的时候,上一个节点是右节点
TreeNode pre=null;
while(root!=null||!stack.isEmpty()){
while(root!=null){
stack.push(root);
root=root.left;
}
//查看栈顶的节点
TreeNode top=stack.peek();
//中间节点遍历过2此,第一次不要打印或记录,第二次需要
//进行判断,即现在是否需要打印或者记录中间(根)节点
//只有当右子树为空,或者右子树已经遍历的时候,才需要进行记录===》第二次
//这里还需要注意一个点,就是这里pop出来的节点的右节点已经遍历,
//所以不能再遍历该节点的右节点,所以这里必须要有一个新的top用来指向
if(top.right==null||top.right==pre){
//只有需要记录的时候才会pop出来
//已经遍历过的时候,拿到该节点
top=stack.pop();
//用pre指向该节点
pre=top;
//记录该节点
res.add(top.val);
}else{
//第一次,啥都不需要,直接遍历右节点
root=top.right;
}
}
return res;
}
}