python（剑指offer38-42题）

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

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

39.输入一棵二叉树，判断该二叉树是否是平衡二叉树。通过判断树的深度来判断。

class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        if pRoot is None:
            return True
        if abs(self.treeDepth(pRoot.left) - self.treeDepth(pRoot.right)) > 1:
            return False
        return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)
        
    def treeDepth(self, root):
        if root is None:
            return 0
        lTree = self.treeDepth(root.left)+1
        rTree = self.treeDepth(root.right)+1
        return max(lTree, rTree)

40.一个整型数组里除了两个数字之外，其他的数字都出现了偶数次。请写程序找出这两个只出现一次的数字。

class Solution:
    # 返回[a,b] 其中ab是出现一次的两个数字
    def FindNumsAppearOnce(self, array):
        # write code here
        li = []
        for i in array:
            if array.count(i) == 1:
                li.append(i)
        return li

41.

题目描述

小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!

输出描述:

输出所有和为S的连续正数序列。序列内按照从小至大的顺序，序列间按照开始数字从小到大的顺序

        # write code here
        ans = []  # 用于存放结果的数组
        juede = int((2 * tsum) ** 0.5)  # 用于判断循环结束的条件
        while juede > 1:
            # 当为奇数时第一个括号判断，为偶数时第二个括号判断
            if (juede&1 and tsum % juede == 0) or (tsum % juede) * 2 == juede:
                li = []  # 用于存储每个符合条件的数组集合
                k = (tsum // juede) - (juede - 1) // 2  # k是满足条件数列的第一个值
                # juede 的值代表这个有序数列的长度
                for j in range(0, juede):
                    li.append(k)
                    k += 1
                ans.append(li)
            juede = juede - 1
        return ans

42.

输入一个递增排序的数组和一个数字S，在数组中查找两个数，使得他们的和正好是S，如果有多对数字的和等于S，输出两个数的乘积最小的。

输出描述: 

对应每个测试案例，输出两个数，小的先输出。通过 in 判断是否存在。

class Solution:
    def FindNumbersWithSum(self, array, tsum):
        # write code here
        ans = []  # 用于存放所有符合的集合
        for i in range(0,tsum+1):
            if i in tsum:
                if (tsum-i) in tsum:
                    li = []
                    li.append(i)
                    li.append(tsum-i)
                    ans.append(li)
        # [] 不等于 None
        if ans == []:
           return []
        return min(ans) # 通过min 函数找到最小的组合