Leetcode 543. 二叉树的直径-----python

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

解题思路：
注意看[4,2,1,3]或[5,2,1,3],是左右两棵子树路径加起来的。

• 二叉树的直径：二叉树中从一个结点到另一个节点最长的路径，叫做二叉树的直径
• 采用分治和递归的思想：根节点为root的二叉树的直径 =
  max(root-left的直径，root->right的直径，root->left的最大深度+root->right的最大深度+1)

python代码实现：

class Solution(object):
def __init__(self):
self.res = 0

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

if root == None: return 0
getMaxDepth(root)
return self.res