day16 二叉树 | 104,559,111,222

LeetCode 104, LeetCode559 求二叉树 or N叉树的最大深度

采用层序遍历的方式

package algor.trainingcamp;

import algor.junior_algor.tree.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

/**
 * @author lizhe
 * @version 1.0
 * @description: 104. 二叉树的最大深度
 *
 * 给定一个二叉树，找出其最大深度。
 *
 * 二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
 *
 * 说明: 叶子节点是指没有子节点的节点。
 *
 * 采用层序法进行遍历，寻找maxDepth
 * @date 2023/4/20 08:05
 */
public class LeetCode104 {
    public int maxDepth(TreeNode root) {
        if(null == root){
            return 0;
        }

        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int depth = 0;

        while(!queue.isEmpty()){
            int len = queue.size();
            //处理单层数据
            depth++;
            while(len > 0){
                TreeNode poll = queue.poll();
                if(poll.left != null){
                    queue.add(poll.left);
                }

                if(poll.right != null){
                    queue.add(poll.right);
                }

                len--;
            }
        }

        return depth;
    }
}

package algor.trainingcamp;

import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @author lizhe
 * @version 1.0
 * @description:
 *
 * 给定一个 N 叉树，找到其最大深度。
 *
 * 最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。
 *
 * N 叉树输入按层序遍历序列化表示，每组子节点由空值分隔（请参见示例）。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/maximum-depth-of-n-ary-tree
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * 同104 使用层序遍历解决
 * @date 2023/4/20 08:13
 */
public class LeetCode559 {
    public int maxDepth(Node1 root) {
        if(null == root){
            return 0;
        }

        Queue<Node1> queue = new LinkedList<>();
        queue.add(root);
        int depth = 0;

        while(!queue.isEmpty()){
            int len = queue.size();
            //处理单层数据
            depth++;
            while(len > 0){
                Node1 poll = queue.poll();
                for(int i = 0;i < poll.children.size();i++){
                    queue.add(poll.children.get(i));
                }

                len--;
            }
        }

        return depth;
    }
}

class Node1 {
    public int val;
    public List<Node1> children;

    public Node1() {}

    public Node1(int _val) {
        val = _val;
    }

    public Node1(int _val, List<Node1> _children) {
        val = _val;
        children = _children;
    }
}

LeetCode111 求二叉树的最小深度

当在层序遍历时，发现第一次遍历到了叶子节点，直接返回深度即可

package algor.trainingcamp;

import algor.junior_algor.tree.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

/**
 * @author lizhe
 * @version 1.0
 * @description:
 *
 * 给定一个二叉树，找出其最小深度。
 * 最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
 *
 * 说明：叶子节点是指没有子节点的节点。
 * @date 2023/4/20 08:18
 */
public class LeetCode111 {
    public int minDepth(TreeNode root) {
        if(null == root){
            return 0;
        }

        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int depth = 0;

        while(!queue.isEmpty()){
            int len = queue.size();
            //处理单层数据
            depth++;
            while(len > 0){
                TreeNode poll = queue.poll();
                if(poll.left != null){
                    queue.add(poll.left);
                }

                if(poll.right != null){
                    queue.add(poll.right);
                }

                //当poll.left 和 poll.right均都为空，达到最小深度 直接弹出
                if(poll.left == null && poll.right == null){
                    return depth;
                }

                len--;
            }
        }

        return depth;
    }
}

LeetCode222

完全二叉树的节点个数，依旧使用层序遍历，但是从原来的计算‘多少层’，变成循环内的计算‘多少节点’

package algor.trainingcamp;

import algor.junior_algor.tree.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

/**
 * @author lizhe
 * @version 1.0
 * @description: 完全二叉树的节点个数
 *
 *  给出一个完全二叉树，求出该树的节点个数。
 * @date 2023/4/20 08:25
 */
public class LeetCode222 {
    public int countNodes(TreeNode root) {
        if(null == root){
            return 0;
        }

        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int result = 0;
        while(!queue.isEmpty()){
            int len = queue.size();

            while(len > 0){
                TreeNode poll = queue.poll();
                result++;
                if(poll.left != null){
                    queue.add(poll.left);
                }

                if(poll.right != null){
                    queue.add(poll.right);
                }

                len--;
            }
        }

        return result;
    }
}