二叉树的后序遍历

101 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第12天,点击查看活动详情

一、题目

LeetCode 二叉树的后序遍历

给你一棵二叉树的根节点 root ,返回其节点值的 后序遍历 。

示例 1:

输入:root = [1,null,2,3]
输出:[3,2,1]

示例 2:

输入:root = []
输出:[]

示例 3:

输入:root = [1]
输出:[1]

提示:

树中节点的数目在范围 [0, 100] 内
-100 <= Node.val <= 100

二、题解

需要对给定的一个二叉树进行后序遍历,后序遍历就是如果左子树不为空的话就先遍历左子树节点,然后如果右子树不为空的话就在遍历右子树节点,最后再遍历到根节点,对于左右子树的节点遍历也是如此。

方法一

可以直接根据后续遍历的规则来用递归方法遍历二叉树的节点,需要需要定义一个数组list来记录后续遍历的结果,然后定义递归方法来遍历二叉树节点,参数首先就需要的是二叉树根节点了,然后还得要一个数组记录遍历的结果,首先判断如果当前参数的树节点为空的话就直接结束递归,说明之后没有节点了,否则就获取根节点的左子树再去递归遍历,然后再获取根节点的右子树去递归遍历,最后再遍历根节点,直接将根节点的值加入到结果数组中,最后递归遍历完成时返回遍历记录的数组即可。

方法二

也可以直接通过一个栈来辅助遍历,首先定义一个栈,然后先将根节点加入栈中,当栈元素不为空的时候遍历栈元素,然后取出栈顶元素节点,如果这个节点为空的话就可以忽略,然后把这个元素根节点值加入到结果数组中最后一位, 如此相当于反向遍历,然后获取这个节点的左子树节点加入到栈中,最后再获取这个节点的右子树节点也加入到栈中,最后遍历完成返回结果即可。

三、代码

方法一 Java代码

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<Integer>();
        dfs(root, list);
        return list;
    }

    public void dfs(TreeNode root, List<Integer> list) {
        if (root == null) {
            return;
        }
        dfs(root.left, list);
        dfs(root.right, list);
        list.add(root.val);
    }
}

时间复杂度:O(n),需要遍历一次二叉树的节点

空间复杂度:O(n),需要使用一个数组记录遍历结果以及递归消耗栈空间。


方法二 Java代码

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        LinkedList<Integer> list = new LinkedList<>();
        if (null == root) {
            return list;
        }
        Deque<TreeNode> deque = new LinkedList<>();
        deque.addFirst(root);
        while (!deque.isEmpty()) {
            TreeNode node = deque.removeFirst();
            if (node == null) {
                continue;
            }
            list.addFirst(node.val);
            deque.addFirst(node.left);
            deque.addFirst(node.right);
        }
        return list;
    }
}

时间复杂度:O(n),需要遍历一次二叉树的节点

空间复杂度:O(n),需要使用一个数组记录遍历结果以及使用栈来辅助遍历。