本博客使用的案例全部来自力扣,并且是由花花酱整理的力扣题
什么是N叉树
N叉树和二叉树的区别就在于:N叉树一个父节点可以拥有多个子节点,而二叉树一个负节点只可以拥有左右两个子节点
N叉树如图所示
N叉树的层序遍历
本题来自力扣429. N 叉树的层序遍历
层序遍历就是将树进行父子分层,然后从腐恶层遍历到子层,并且是从左到右遍历。
因此下图的分层结果是 1 234 56
仔细思考,我们可以使用队列的方法来解决。
我们先将头节点放入队列中
然后将头节点抛出,使用一个节点对象接收,然后将节点值放入到list,同时将子节点全部放入队列中
再将节点抛出,并检查它的是否有子节点,有的话就加入
直到暑促到tai就开始遍历子节点,并且时按顺序来的
class Solution {
public List<List<Integer>> levelOrder(Node root) {
List<List<Integer>> lists = new ArrayList<List<Integer>>();
if(root == null){
return lists;
}
Queue<Node> queue = new LinkedList<Node>();
queue.add(root);
while(!queue.isEmpty()){
List<Integer> levelList = new ArrayList<>();
int size = queue.size();
for(int i = 0; i < size; i++){
Node node = queue.poll();
levelList.add(node.val);
queue.addAll(node.children);
}
lists.add(levelList);
}
return lists;
}
}
N叉树的前序遍历
本题来自力扣589. N 叉树的前序遍历
N叉树的前序遍历的规则和二叉树的前序遍历的规则时一样的,如下图所示的N叉树前序遍历的结果是:125634
和二叉树有所不同的是孩子有多个不能像二叉树一样先遍历右子树,后遍历左子树,这是一个要解决的小问题,再综合题目给我们的子树类型是 List 类型,因此我们可以使用0-n来确定子树的位置。
解决代码如下
class Solution {
public List<Integer> preorder(Node root) {
List<Integer> result = new ArrayList();
if(root == null){
return result;
}
recursion(root,result);
return result;
}
public void recursion(Node root, List<Integer> result){
if(root != null){
result.add(root.val);
for(int i = 0; i < root.children.size(); i++){
recursion(root.children.get(i), result);
}
}
}
}
和二叉树一样,这个也是可以使用迭代的方式替代递归,下面是来自力扣官网的题解代码,这里我会使用图解将这个过程演示出来
class Solution {
public List<Integer> preorder(Node root) {
LinkedList<Integer> output = new LinkedList<>();
if (root == null) {
return output;
}
LinkedList<Node> stack = new LinkedList<>();
stack.add(root);
while (!stack.isEmpty()) {
Node node = stack.pollLast();
output.add(node.val);
Collections.reverse(node.children);
for (Node item : node.children) {
stack.add(item);
}
}
return output;
}
}
作者:LeetCode
链接:https://leetcode-cn.com/problems/n-ary-tree-preorder-traversal/solution/ncha-shu-de-qian-xu-bian-li-by-leetcode/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
在进入while循环前
Collections.reverse(node.children);
是将子树排序翻转因此在for循环执行结束后,栈和队列的值为:
1 已经被栈去顶去掉了,然后向栈中逆序压入1 的子树
再进行一轮while循环以后为
2 也被栈去顶去掉了,然后逆序加入子树,再后面就是不停的弹栈,这个算法就结束了
N叉树的后序遍历
本题来自力扣590. N 叉树的后序遍历
后序遍历和二叉树的也是一样,要解决的问题也是和前序遍历一样的问题。这里就不过多赘述,结果代码如下
class Solution {
public List<Integer> postorder(Node root) {
List<Integer> result = new ArrayList();
if(root == null){
return result;
}
recursion(root, result);
return result;
}
public void recursion(Node root, List<Integer> result){
if(root != null){
for(int i = 0; i < root.children.size(); i++){
recursion(root.children.get(i), result);
}
result.add(root.val);
}
}
}