剑指offer 和为S的两个数字 python

版权声明:本文为博主原创文章,需转载可以私信我,同意后即可 https://blog.csdn.net/Sun_White_Boy/article/details/83539131

题目描述

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

样例

输入[1,2,3,4,10],11
输出[1,10]

想法一:
由于是排序好的数组,所以从最小的数字low开始找,在数组中找是否存在tsum-low即可

class Solution:
    def FindNumbersWithSum(self, array, tsum):
        if not array:
            return []
        for i in range(len(array)-1):
            low = array[i]
            if low >= tsum//2:
                break
            try:
                array.index(tsum-low)
                return [low, tsum-low]
            except ValueError:
                continue
        return []

想法二:
头尾指针方法,如果两个和大于tsum,那么说明尾指针值过大,则尾指针左移,如果和小于tsum,则说明头指针值过小,则头指针右移,如果等值,则返回[头,尾].

class Solution:
    def FindNumbersWithSum(self, array, tsum):
        i, j = 0, len(array)-1
        while i < j:
            if array[i] + array[j] == tsum:
                return [array[i], array[j]]
            while i < j and array[i] + array[j] > tsum:
                j -= 1
            while i < j and array[i] + array[j] < tsum:
                i += 1
        return []


最后

刷过的LeetCode或剑指offer源码放在Github上了,希望喜欢或者觉得有用的朋友点个star或者follow。
有任何问题可以在下面评论或者通过私信或联系方式找我。
联系方式
QQ:791034063
Wechat:liuyuhang791034063
CSDN:https://blog.csdn.net/Sun_White_Boy
Github:https://github.com/liuyuhang791034063

猜你喜欢

转载自blog.csdn.net/Sun_White_Boy/article/details/83539131