Leetcode刷题之旅--559. N叉树的最大深度

最新推荐文章于 2024-10-31 16:16:13 发布

祥哲

最新推荐文章于 2024-10-31 16:16:13 发布

阅读量125

点赞数

文章标签： leetcode 算法 java 二叉树

本文链接：https://blog.csdn.net/weixin_45129136/article/details/106658484

版权

题目描述：
在这里插入图片描述

思路:和二叉树的深度一样，只不过换成了n叉树。左右孩子换成了一个孩子列表。
题目给的结点定义

import java.util.List;

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

解决：

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

class Solution {
    public int maxDepth(Node root) {
        if (root==null) return 0;
        if (root.children.isEmpty()) return 1;
        List height=new LinkedList();
        int maxhei=0;
        int hei=0;
        for (int i=0;i<root.children.size();i++){
            hei=maxDepth(root.children.get(i));
            height.add(hei);
            maxhei=hei>maxhei?hei:maxhei;
        }
        return maxhei+1;
    }
}

此外还有迭代和栈来解决的方法。
附上官方迭代+栈的解法。

import javafx.util.Pair;
        import java.lang.Math;

class Solution {
    public int maxDepth(Node root) {
        Queue<Pair<Node, Integer>> stack = new LinkedList<>();
        if (root != null) {
            stack.add(new Pair(root, 1));
        }

        int depth = 0;
        while (!stack.isEmpty()) {
            Pair<Node, Integer> current = stack.poll();
            root = current.getKey();
            int current_depth = current.getValue();
            if (root != null) {
                depth = Math.max(depth, current_depth);
                for (Node c : root.children) {
                    stack.add(new Pair(c, current_depth + 1));
                }
            }
        }
        return depth;
    }
};