牛客66道编程题——二叉树的深度

题目

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

思路

求树的深度，可以从层序遍历去考虑。
层序遍历可以使用队列完成，也可以用递归完成。

1、用队列完成

（1）判断边界条件:树是否为空，若为空则返回空
（2）定义一个列表q来存储该层的结点，并且将该列表初始化为树的根结点
（3）定义一个常数count来确定树的深度
（4）建立一个while循环来确定树的深度，循环的依据是q的长度不为0，因为q的长度为0的话，说明这个数已经遍历完毕，无需再遍历。

在这个遍历中：
1）建立一个空的列表tmp来存储同层的结点
2）获取q的长度
3）建立一个for循环来判断q中的结点是否有左结点和右结点。
4）若tmp不为0，说明这一层中右结点，则count加1

（5）循环结束后，count就是这棵树的深度。

class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        if pRoot==None:#判断边界条件
            return 0
        q=[]#定义一个列表存储当层结点
        q.append(pRoot)#并把该列表初始化为树的根结点
        count=0#定义count为0来确定树的深度
        while len(q)!=0:#创立一个while循环来确定二叉树的深度
            tmp=[]
            l=len(q)
            for i in range(l):#创建一个for循环来判断该结点中是否还有左结点和右节点
                r=q.pop(0)
                if r.left!=None:
                    q.append(r.left)
                if r.right!=None:
                    q.append(r.right)
                tmp.append(r.val)
            if tmp:
                count+=1
        return count

2、递归法

class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        if pRoot==None:
            return 0
        count=max(self.TreeDepth(pRoot.left),self.TreeDepth(pRoot.right))+1
        return count