Leetcode刷题589. N叉树的前序遍历

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

例如，给定一个 3叉树 :

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

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

class Solution {
    public List<Integer> preorder(Node root) {
//        return preorderI(root);
        return preorderII(root);
    }

    //方法二：迭代，时间和空间复杂度O(N)
    private List<Integer> preorderII(Node root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }
        //先将根节点入栈
        Stack<Node> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            root = stack.pop();
            list.add(root.val);
            //前序遍历根左右，入栈时先将右节点入栈
            for (int i = root.children.size() - 1; i >= 0; i--) {
                stack.push(root.children.get(i));
            }
        }
        return list;
    }

    //方法一：递归，时间和空间复杂度O(N)
    private List<Integer> preorderI(Node root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }
        preOrder(root, list);
        return list;
    }

    private void preOrder(Node root, List<Integer> list) {
        if (root == null) {
            return;
        }
        //前序遍历根左右，这里先将根节点添加到结果集，然后再处理N叉树的子节点
        list.add(root.val);
        for (Node node : root.children) {
            preOrder(node, list);
        }
    }
}