Leetcode590.N叉树的后序遍历

题目描述

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

示例：

返回其后序遍历: [5,6,3,2,4,1].

题解

头插法后序遍历（java）

思路：对树进行遍历，同时用头插来保证后序遍历。(栈 + 头插)

/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;
    public Node() {}
    public Node(int _val) {
        val = _val;
    }
    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
    public List<Integer> postorder(Node root) {
        LinkedList<Node> stack = new LinkedList<>();
        LinkedList<Integer> output = new LinkedList<>();
        if (root == null) {
            return output;
        }
      stack.add(root);
      while (!stack.isEmpty()) {
          Node node = stack.pollLast();
          output.addFirst(node.val);
          for (Node item : node.children) {
              if (item != null) {
                  stack.add(item);    
              } 
          }
      }
      return output;
    }
}

复杂度分析

• 时间复杂度： $O(N)$
• 空间复杂度： $O(N)$

递归（java）

思路：

class Solution {
    public List<Integer> postorder(Node root) {
        List<Integer> res  =  new ArrayList<Integer>();
        if(root == null) return res;
        for(Node cur:root.children){
            res.addAll(postorder(cur));
        }
        res.add(root.val);
        return res;
    }
}

复杂度分析

• 时间复杂度： $O(N)$
• 空间复杂度： $O(N)$