劍指offer 和為S的兩個數字 python
阿新 • • 發佈:2018-12-26
題目描述
輸入一個遞增排序的陣列和一個數字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