牛客网 二叉树的深度

题目：

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。

解答：

找到二叉树中的所有路径，然后获取最长的路径的长度。

找到二叉树中所有路径的方法参照：牛客网 二叉树中和为某一值的路径

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def TreeDepth(self, pRoot):
        if not pRoot:
            return 0
        ans=self.TreePath(pRoot)
        # 获取长度
        length=0
        for i in ans:
            if length<len(i):
                length=len(i)
        return length

    def TreePath(self,pRoot):
        if not pRoot:
            return []
        if pRoot and not pRoot.left and not pRoot.right:
            return [[pRoot.val]]

        ans=[]
        left=self.TreePath(pRoot.left)
        right=self.TreePath(pRoot.right)
        for i in left+right:
            ans.append([pRoot.val]+i)

        return ans