543.二叉树的直径--python

题：给定一棵二叉树，计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。例如题目中给定二叉树，它的直径长度为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))