LeetCode.589. N 叉树的前序遍历

题目

589. N 叉树的前序遍历

分析

我们之前有做过LeetCode的 144. 二叉树的前序遍历，其实对于 N 插树来说和二叉树的思路是一模一样的。
二叉树的前序遍历是【根 左 右】
N叉树的前序遍历顺序是【根 孩子】，你可以把二叉树的【根 左 右】想象成【根 孩子】，因为左右就是孩子。

代码

先看一下二叉树前序遍历的代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        func(root,res);
        return res;
    }

    void func(TreeNode cur,List<Integer> res) {
        if(cur == null) return;
        // 先记录根节点
        res.add(cur.val);
        // 遍历左子树
        func(cur.left,res);
        // 遍历右子树
        func(cur.right,res);
    }
}

只需要改动一点就是N叉树的前序遍历代码，如下：

/*
// 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> preorder(Node root) {
        List<Integer> res = new ArrayList<>();
        if(root == null) return res;
        pre(root,res);
        return res;
    }

    void pre(Node cur,List<Integer> res) {
        if(cur == null) return ;
        // 先记录当前节点
        res.add(cur.val);
        // 再遍历孩子
        for(Node node : cur.children) {
            pre(node,res);
        }
    }
}