【Python实践-8】和为S的两个数字

发布时间:2019-05-31 22:12:08编辑:auto阅读(1932)

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

    思路:选定第一个数字,然后遍历后面的数字求和并与S比较,需要n-1次,不行的话再选定第2,3,,,n个数字,需要n^2次,时间复杂度比较高。更简单的方法可以是定义两个指针,第一个指向第一个元素,第二个指向最后一个元素,两个元素相加,如果等于S则输出这两个元素,如果大于,则将第二个指针向前移一位,再求和进行比较;如果小于,则将第一个指针向前移一位,再进行求和比较。

     1 def findNumberWithSum(data,tsum):
     2     i=0
     3     j=len(data)-1
     4     if not data or not tsum:
     5         return []
     6     while i<len(data) and j>0:
     7         if data[i]+data[j]==tsum:
     8             return (data[i],data[j])
     9         if data[i]+data[j]>tsum:
    10             j=j-1
    11         if data[i]+data[j]<tsum:
    12             i=i+1
    13         return ()

    知识点:

    1、if not x是判断是否为None的情况,if x is not None这种写法也是可以的。注意代码完备性,需判断传入参数是否为空。

    2、涉及到两个元素,想到定义两个指针,避免多层循环。

    3、要考虑找不到两个数的情况,可以输出一个空列表或空元组。

关键字