543.二叉树的直径--python

    xiaoxiao2022-06-30  96

    题:给定一棵二叉树,计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。例如题目中给定二叉树,它的直径长度为4。

    1 / \ 2 3 / \ 4 5

    法:这题用一个辅助函数计算当前节点的左右子树的depth,然后depth_left+depth_right就是当前节点的路径长度。一开始写这道题我犯了个错误,认为最大路径一定是通过根节点,所以就算了根节点的左右子数的depth,但其实不一定~~

    注意:一开始要判定root是否为None,为None的话直接return 0,否则root.left和root.right不存在,程序会报错。

    def diameterOfBinaryTree(self, root): """ :type root: TreeNode :rtype: int """ if root==None:return 0 res=self.depth(root.left)+self.depth(root.right) return max(res,self.diameterOfBinaryTree(root.left),self.diameterOfBinaryTree(root.right)) def depth(self,root): if root==None: return 0 return 1+max(self.depth(root.left),self.depth(root.right))

     


    最新回复(0)