N叉树的遍历
在学习数据结构时,我们都了解了二叉树的遍历,包括前序遍历,中序遍历,后序遍历,以下是前序遍历的示例代码
// 前序遍历根节点为root的二叉树,并将遍历结果存储在List中
class Solution{
List<Integer> ans=new ArrayList<>();
public List<Integer> preorder(TreeNode root){
if(root==null) return ans;
ans.add(root.val);
preorder(root.left);
preorder(root.right);
return ans;
}
}
简单的,对于中序和后序遍历只需要改变其中几行代码的顺序即可
// 中序
preorder(root.left);
ans.add(root.val);
preorder(root.right);
// 后序
preorder(root.left);
preorder(root.right);
ans.add(root.val);
做到这,我们只是知道了二叉树的遍历。拓展一下,如果是一个N叉树呢?每个节点不一定只有两个分支,我们应该如何进行这三种遍历?
例:
给定一个 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]
事实上,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 {
List<Integer> ans=new ArrayList<>();
public List<Integer> preorder(Node root) {
if(root==null) return ans;
ans.add(root.val);
for(Node node:root.children){
preorder(node);
}
return ans;
}
}
END