leetcode二叉树-二叉树的最大深度 dfs、bfs

package binarytree.maxDepth;

import binarytree.untils.GenerateTreeNode;
import binarytree.untils.TreeNode;

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

/**
 * 104. 二叉树的最大深度
 * 给定一个二叉树,找出其最大深度。
 * <p>
 * 二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
 * <p>
 * 说明: 叶子节点是指没有子节点的节点。
 * <p>
 * 示例:
 * 给定二叉树 [3,9,20,null,null,15,7],
 * <p>
 * 3
 * / \
 * 9  20
 * /  \
 * 15   7
 * 返回它的最大深度 3 。
 */
public class maxDepth {
    /**
     * 基本功 层序遍历即可
     *bfs
     * @param root
     * @return
     */
    public static int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        return levelOrder(root);
    }

    private static int levelOrder(TreeNode root) {
        List<List<Integer>> re = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList();
        queue.offer(root);
        while (!queue.isEmpty()) {
            List<Integer> res = new ArrayList<>();
            int size = queue.size();

            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                res.add(node.val);
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }

            }
            if (res.size() != 0) {
                re.add(res);
            }
        }
        return re.size();
    }


    //dfs
    public static int maxDepth2(TreeNode root) {
        if (root == null) {
            return 0;
        }
        if(root.left == null && root.right==null){
            return 1;
        }

        int l = maxDepth2(root.left);
        int r = maxDepth2(root.right);

        if(root.left==null || root.right==null){
            return l+r+1;
        }
        return Math.max(l,r)+1;
    }

    public static void main(String[] args) {
        Integer[] nums = {1, 2, 3, 4, 5, 6, 7};
        TreeNode treeNode = GenerateTreeNode.generateTreeNode(nums);

        System.out.println(maxDepth2(treeNode));
    }
}

posted @ 2021-12-15 10:38  小傻孩丶儿  阅读(27)  评论(0编辑  收藏  举报