LeetCode刷题-N叉树的最大深度

前言说明

算法学习,日常刷题记录。

题目连接

N叉树的最大深度

题目内容

给定一个N叉树,找到其最大深度。

最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。

N叉树输入按层序遍历序列化表示,每组子节点由空值分隔(请参见示例)。

示例1:

图1

输入:root = [1,null,3,2,4,null,5,6]

输出:3

示例2:

图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]

输出:5

提示:

树的深度不会超过1000。

树的节点数目位于[0, 10^4]之间。

分析过程

思路:深度优先搜索,利用递归,若一棵树有n个节点,设maxChildDepth为n个节点中的最大深度,那么这棵树的最大深度是n+1。

第一步

若节点为空,返回0。

第二步

定义最大子节点深度maxChildDepth,初始为0;获取节点的子节点children,遍历子节点children。

第三步

每次遍历时,递归调用自身函数maxDepth,获取每一个子节点的最大深度,定义为childDepth,和当前最大子节点深度maxChildDepth比较,取最大值,更新到最大子节点深度maxChildDepth。

第四步

最后,返回最大子节点深度maxChildDepth加1,即为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 int maxDepth(Node root) {
        // 深度优先搜索,若一棵树有n个节点,设maxChildDepth为n个节点中的最大深度,那么这棵树的最大深度是n+1

        if (root == null) {
            // 若节点为空,返回0
            return 0;
        }

        // 定义最大子节点深度
        int maxChildDepth = 0;

        // 获取节点的子节点
        List<Node> children = root.children;

        if (children != null) {
            // 遍历子节点
            for (Node node : children) {
                // 递归调用自身,获取每一个子节点的最大深度
                int childDepth = maxDepth(node);

                // 和当前最大子节点深度比较,取最大值,更新到最大子节点深度
                maxChildDepth = Math.max(maxChildDepth, childDepth);
            }
        }

        // 返回最大子节点深度加1,即为n叉树的最大深度
        return maxChildDepth + 1;
    }
}

提交结果

执行用时1ms,时间击败100.00%的用户,内存消耗38.7MB,空间击败10.62%的用户。

运行结果

延伸分析

但是,题目中并没有给出输入的数组转为N叉树的代码,在LeetCode上刷题时LeetCode后台已经帮你做了转换,但是如果我们要自己写出完整代码,需要自己做转换,思路和之前的数组转二叉树类似,通过队列辅助来实现广度优先搜索,即模拟层序遍历时从上到下从左到右的过程,把数组转换为N叉树,不过这里不同的是,每组子节点由空值分隔,改为遇到空值时才进行队列出列操作。

数组转二叉树,请看文章:LeetCode刷题-我会翻转二叉树,谷歌还要我吗?

完整代码如下:

import java.util.*;

public class Main {

    public static void main(String[] args) {
        // 获取输入结果
        Scanner scanner = new Scanner(System.in);
        String str = scanner.next();
        scanner.close();

        // 处理输入结果
        Integer[] nums;
        if ("[]".equals(str)) {
            // 当数组为空
            nums = null;
        } else {
            // 当数组不为空
            String[] strs = str.split("\\[")[1].split("]")[0].split(",");
            int size = strs.length;
            nums = new Integer[size];
            for (int i = 0; i < size; ++i) {
                if ("null".equals(strs[i])) {
                    nums[i] = null;
                } else {
                    nums[i] = Integer.parseInt(strs[i]);
                }
            }
        }

        // 数组转为N叉树
        Node root = arrayToTreeNode(nums);

        // 获取输出结果
        int result = maxDepth(root);
        System.out.println(result);
    }

    // 数组转为N叉树,层序遍历,分层按照从上到下,从左到右的顺序
    private static Node arrayToTreeNode(Integer[] nums) {
        // 利用队列辅助,N叉树输入按层序遍历序列化表示,每组子节点由空值分隔
        // 例子:[1,null,3,2,4,null,5,6],[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]

        if (nums == null || nums.length == 0) {
            // 若N叉树节点为空,直接返回空
            return null;
        }

        // 数组的开始下标
        int i = 1;

        // 先构造N叉树根节点
        Node root = new Node(nums[0]);

        // 定义当前的N叉树节点,保存临时值
        Node current = null;

        // 定义当前数组的值
        Integer value;

        // 定义队列,层序遍历创建N叉树
        Queue<Node> queue = new LinkedList<>();

        // 先把N叉树根节点放进队列最后
        queue.add(root);

        // 遍历数组,构造N叉树
        while (i < nums.length) {
            if (nums[i] == null) {
                // 若节点值为空,下标加1
                ++i;

                // 若节点值为空,队列出列元素,获取当前父节点
                current = queue.poll();
            } else {
                // 获取数组的值,数组下标加1
                value = nums[i++];

                // 创建当前父节点的孩子
                Node node = new Node(value);

                if (current == null) {
                    current = new Node();
                }

                if (current.children == null) {
                    current.children = new ArrayList<>();
                }

                // 当前父节点添加孩子
                current.children.add(node);

                // 把当前父节点的孩子入列
                queue.add(node);
            }
        }

        // 返回N叉树的根节点
        return root;
    }

    private static int maxDepth(Node root) {
        // 深度优先搜索,若一棵树有n个节点,设maxChildDepth为n个节点中的最大深度,那么这棵树的最大深度是n+1

        if (root == null) {
            // 若节点为空,返回0
            return 0;
        }

        // 定义最大子节点深度
        int maxChildDepth = 0;

        // 获取节点的子节点
        List<Node> children = root.children;

        if (children != null) {
            // 遍历子节点
            for (Node node : children) {
                // 递归调用自身,获取每一个子节点的最大深度
                int childDepth = maxDepth(node);

                // 和当前最大子节点深度比较,取最大值,更新到最大子节点深度
                maxChildDepth = Math.max(maxChildDepth, childDepth);
            }
        }

        // 返回最大子节点深度加1,即为n叉树的最大深度
        return maxChildDepth + 1;
    }

}

// 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;
    }
};

其中,arrayToTreeNode方法就是把输入的数组转为N叉树。

完整代码获取:https://github.com/zjhpure/algorithmPractice/blob/master/src/main/java/org/pure/algorithm/maximumDepthOfNAryTree/Main.java

原文链接

原文链接:N叉树的最大深度

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,458评论 4 363
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,454评论 1 294
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,171评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,062评论 0 207
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,440评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,661评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,906评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,609评论 0 200
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,379评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,600评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,085评论 1 261
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,409评论 2 254
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,072评论 3 237
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,088评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,860评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,704评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,608评论 2 270

推荐阅读更多精彩内容