leetcode 590. N 叉树的后序遍历

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

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]
 

提示:

N 叉树的高度小于或等于 1000
节点总数在范围 [0, 10^4] 内

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

后序遍历有多种方式,迭代,递归,这道题还可以先前序遍历,之后反转集合。采用迭代方式,用两个栈来,一个记录节点,另一个记录是否输出。

      public List<Integer> postorder(Node root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }
        Stack<Node> stack = new Stack<>();
        Stack<Boolean> flag = new Stack<>();
        stack.push(root);
        flag.push(false);
        while (!stack.isEmpty()) {
            Boolean f = flag.pop();
            if (f) {
                Node node = stack.pop();
                list.add(node.val);
            } else {
                flag.push(true);
                Node node = stack.peek();
                List<Node> children = node.children;
                if (children != null && !children.isEmpty()) {
                    for (int i = children.size() - 1; i >= 0 ; i--) {
                        stack.push(children.get(i));
                        flag.push(false);
                    }
                }
            }
           
        }
        return list;
    }

运行之后,发现比较耗时,这道题还是建议采用递归或者前序遍历反转集合比较好。

posted @ 2021-04-19 13:24  旺仔古李  阅读(48)  评论(0编辑  收藏  举报