LeetCode_589_N叉树的前序遍历

题目链接

• https://leetcode.cn/problems/n-ary-tree-preorder-traversal/

题目描述

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

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

示例 1：

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

示例 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]
输出：[1,2,3,6,7,11,14,4,8,12,5,9,13,10]

提示：

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

进阶：递归法很简单，你可以使用迭代法完成此题吗?

解题思路

迭代法

• 二叉树的前序遍历是中左右，每次先处理的是中间节点，那么先将根节点放入栈中，然后将右孩子送进栈，最后将左孩子送进栈。这样出栈的时候就是中左右的顺序
• N叉树中没有可能有多个孩子，那么将从右到左依次将孩子入栈，这样出栈的时候就是中左…右

AC代码

class Solution {
    public List<Integer> preorder(Node root) {
        List<Integer> ans = new ArrayList<>();
        if(root == null){
            return ans;
        }
        Stack<Node> stack = new Stack<>();
        stack.push(root);
        while(!stack.isEmpty()){
            Node tmp = stack.pop();
            ans.add(tmp.val);
            for(int i = tmp.children.size() -1; i >= 0; i--){
                stack.push(tmp.children.get(i));
            }
        }
        return ans;
    }
}