[LeetCode] 590. N-ary Tree Postorder Traversal

原题链接： https://leetcode.com/problems/n-ary-tree-postorder-traversal/

1. 题目介绍

Given an n-ary tree, return the postorder traversal of its nodes’ values.
给出一个n叉树，返回后序遍历的结果。
后序遍历就是指先从左到右遍历完所有的子树，然后再遍历根节点。
For example, given a 3-ary tree:
比如下面这个三叉树：

Return its postorder traversal as: [5,6,3,2,4,1].
后序遍历顺序为[5,6,3,2,4,1].

Note:
Recursive solution is trivial, could you do it iteratively?
递归的算法太简单了，要不要试一试迭代？

2. 解题思路

2.1 递归

递归的方法就是，先遍历所有的子树，再将根节点的 val 值添加进List中去。
实现代码

/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val,List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
    public List<Integer> postorder(Node root) {
        List<Integer> l = new ArrayList<>();
        return helper(root,l);
    }
    public List<Integer> helper(Node root,List<Integer> l){
        if(root == null){
            return l;
        }
        for(Node i : root.children){
            helper(i,l);
        }
        l.add(root.val);
        return l;
    }    
}

2.2 迭代

使用栈进行辅助遍历。
当遍历到一个节点时，如果该节点不为空，那么就首先新建一个和该节点val值一样，但是children值为空的新节点放入堆栈。
然后从后向前依次将子树节点放入堆栈。剩下第一个子树不要放入堆栈，继续遍历第一个子树。
如果该节点为空，那么就从堆栈顶部弹出一个节点，继续遍历弹出的节点。
实现代码

/*
// Definition for a Node.
class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val,List<Node> _children) {
        val = _val;
        children = _children;
    }
};
*/
class Solution {
    public List<Integer> postorder(Node root) {
        
        List<Integer> l = new ArrayList<>();
        Stack<Node> s = new Stack<>();
        
        Node temp = root;
        while(temp != null || s.isEmpty() != true){
            if(temp != null){
                int length = temp.children.size();
                if(length == 0){
                    l.add(temp.val);
                    temp = null;
                    continue;
                }else{
                    s.push( new Node( temp.val , new ArrayList<>() ) );
                    for(int i = length-1 ; i > 0 ; i--){
                        s.push(temp.children.get(i));
                    }
                    temp = temp.children.get(0);
                }

            }else{
                temp = s.pop();
            }
        }
        return l;
    }
}