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