欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

543. 二叉树的直径

程序员文章站 2022-03-03 10:34:59
...

给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    

返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意:两结点之间的路径长度是以它们之间边的数目表示。

class Solution(object):

    def diameterOfBinaryTree(self, root):
        """
        :type root: TreeNode
        :rtype: int
         1
         / \
        2   3
       / \
      4   5
        """
        self.cur=0
        self.l=0
        if root==None:
            return  0
        # 中序遍历
        def inorder(root):
            if not root:

                return 0
            
            temp1=inorder(root.left)
            temp2=inorder(root.right)

            if temp2+temp1>self.cur:
                self.cur=temp2+temp1

            if (temp1)>=(temp2):
                return  temp1+1

            return temp2+1

        inorder(root)
        return  self.cur