二叉树的深度
题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
解题思路:
利用递归实现。如果一棵树只有一个结点,那么它的深度为1。递归的时候无需判断左右子树是否存在,因为如果该节点
为叶节点,它的左右子树不存在,那么在下一级递归的时候,直接return 0。同时,记得每次递归返回值的时候,深度加一操
作,因为计算深度是从根节点下面一个节点开始计算的。
代码:
# -*- 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
return max(self.TreeDepth(pRoot.left),self.TreeDepth(pRoot.right))+1
平衡二叉树
题目描述
输入一棵二叉树,判断该二叉树是否是平衡二叉树。
解题思路:
如果二叉树的每个节点的左子树和右子树的深度不大于1,它就是平衡二叉树。
先写一个求深度的函数,再对每一个节点判断,看该节点的左子树的深度和右子树的深度的差是否大于1
代码:
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def IsBalanced_Solution(self, pRoot):
# write code here
if pRoot == None:
return True
depth1 = self.GetDepth(pRoot.left)
depth2 = self.GetDepth(pRoot.right)
if abs(depth1-depth2)>1:
return False
return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)
def GetDepth(self, root):
if not root:
return 0
return max(self.GetDepth(root.left), self.GetDepth(root.right))+1
数组中只出现一次的数字
题目描述
一个整型数组里除了两个数字之外,其他的数字都出现了偶数次。请写程序找出这两个只出现一次的数字。
解题思路:
用python自带的Counter库。返回一个列表,map(f,input),对input进行f操作,第一个参数lambda函数,意思取返回值中的第一个数,因为counter函数返回的是字典,counter().most_common返回的是有序的计数字段,取最后两个,顺序是从大到小的
list(map(lambda c:c[0],Counter(array).most_common()[-2:]))
代码:
# -*- coding:utf-8 -*-
class Solution:
# 返回[a,b] 其中ab是出现一次的两个数字
def FindNumsAppearOnce(self, array):
# write code here
from collections import Counter
return list(map(lambda c:c[0], Counter(array).most_common()[-2:]))