Leetcode 1522. Diameter of N-Ary Tree (python+cpp)

努利！奋斗！

于 2020-09-07 06:41:54 发布

阅读量452

点赞数

分类专栏： Leetcode 树文章标签： leetcode python c++

本文链接：https://blog.csdn.net/qq_37821701/article/details/108440078

版权

Leetcode 树专栏收录该内容

35 篇文章

订阅专栏

本文介绍了LeetCode问题1522的解决方案，主要探讨如何找到一棵N-叉树的最长路径，即直径。解法与543题类似，但需找出所有子树中深度最大的两个。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

在这里插入图片描述

解法：

这是543的follow up，解法几乎是一样的，只不过左子树和右子树的深度替换成所有子树里面深度最大的两个

class Solution:
    def diameter(self, root: 'Node') -> int:
        """
        :type root: 'Node'
        :rtype: int
        """
        def get_depth(node):
            nonlocal diameter
            largest = 0
            sec_largest = 0
            for child in node.children:
                d = get_depth(child)
                if d>largest:
                    sec_largest = largest
                    largest = d
                elif d>sec_largest:
                    sec_largest = d
            diameter = max(diameter,largest+sec_largest)
            return max(largest,sec_largest)+1
        
        diameter = 0
        get_depth(root)
        return diameter

C++

class Solution {
    int ans;
public:
    int diameter(Node* root) {
        ans = 0;
        get_depth(root);
        return ans;
    }
    int get_depth(Node* node){
        if (!node) return 0;
        int largest = 0;
        int sec_largest = 0;
        
        for (auto child:node->children){
            int d = get_depth(child);
            if (d>largest){
                sec_largest = largest;
                largest = d;
            }
            else if (d>sec_largest){
                sec_largest = d;
            }
        }
        ans = max(ans,largest+sec_largest);
        return max(largest,sec_largest)+1;
        
    }
};