N 叉树的后序遍历

205 阅读1分钟

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

一、题目

LeetCode N 叉树的后序遍历

给定一个 n 叉树的根节点 root ,返回 其节点值的 后序遍历 。

n 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。

示例 1:

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

示例 2:

输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[2,6,14,11,7,3,12,8,4,13,9,10,5,1]

提示:

节点总数在范围 [0, 104]0 <= Node.val <= 104
n 叉树的高度小于或等于 1000

二、题解

需要对给定的树进行后序遍历,而树节点是不定的,后序遍历就是如果左子树不为空的话就先遍历左子树,如果右子树不为空的话就遍历右子树,最后在遍历根节点,对于子树也是如此来遍历。

方法一

对此可以通过使用DFS深度优先搜索,从树的根节点开始遍历,然后对子树进行递归遍历。具体的定义dfs递归函数,参数需要一个root树的根节点来遍历,然后还需要一个记录遍历结果的数组,递归内判断如果当前节点为空的话就结束遍历,说明之后无节点不需遍历了,否则就获取当前节点的子树,先遍历子树,因为子树是不定的,所以给定的子树是给数组存储,那么可以直接对子树数组顺序遍历,将每一个子树再去递归遍历,这样就达到了先从左到右遍历子树,最后遍历完子树再遍历根节点,将根节点的值加入到记录数组中。最后遍历结束的时候返回记录的数组即可。

方法二

或者也可以先通过队列辅助遍历,利用队列模拟递归遍历,但是需要注意遍历过程中可能重复,所以遍历过程中需要记录遍历节点,如果当前节点以及遍历够就需要忽略,或者遍历节点是叶子节点了也忽略,否则就继续遍历记录遍历结果。

三、代码

方法一 Java代码

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

    public void dfs(Node root, List<Integer> list) {
        if (root == null) {
            return;
        }
        for (Node node : root.children) {
            dfs(node, list);
        }
        list.add(root.val);
    }
}

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

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