剑指offer——38二叉树的深度(Python)

【题目】输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。
【思路】求二叉树的深度可以通过求其子树的深度加1即可，在左子树和右子树子树中选择最大的。
【代码】

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        if pRoot == None:
            return 0
        
        left = self.TreeDepth(pRoot.left)
        right = self.TreeDepth(pRoot.right)
        if left > right:
            return left+1
        else:
            return right+1

还可以对树进行层次遍历，首先将根节点存入队列，然后取出根节点，在队列中加入根节点的左右子节点，按层次进行取放，每存一层，层数就加1，最后没有子节点存了，队列为空了，结束循环。

class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        # 层数
        count = 0
        # 如果根节点为空，则返回空列表
        if pRoot is None:
            return count
        # 模拟一个队列储存节点
        q = []
        # 首先将根节点入队
        q.append(pRoot)
        # 列表为空时，循环终止
        while len(q) != 0:
            # 记录同层节点的个数
            length = len(q)
            for i in range(length):
                # 将同层节点依次出队
                r = q.pop(0)
                if r.left is not None:
                    # 非空左孩子入队
                    q.append(r.left)
                if r.right is not None:
                    # 非空右孩子入队
                    q.append(r.right)
            count += 1  # 统计层数
        return count