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);}}
}