38.二叉树的深度(python)

题目描述

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

 

思路：用层次遍历，每次遍历到每层的最后一个节点，高度增加1

class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        if pRoot==None:
            return 0
        queue = [pRoot]
        last=0
        h = 0
        index=0
        while index<len(queue):
            tmpRoot=queue[index]
            if tmpRoot.left!=None:
                queue.append(tmpRoot.left)
            if tmpRoot.right!=None:
                queue.append(tmpRoot.right)
            if index==last:
                h+=1
                last=len(queue)-1
            index+=1
        return h

2019-12-25 16:23:21

递归版

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

2019-12-25 16:46:14