1. 程式人生 > >劍指offer 和為S的兩個數字 python

劍指offer 和為S的兩個數字 python

題目描述

輸入一個遞增排序的陣列和一個數字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