和为S的两个数字

###题目

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

解题思路

设两个头尾两个指针i和j:
若ai + aj == sum,就是答案(相差越远乘积越小);
若ai + aj > sum,aj肯定不是答案之一(前面已得出 i 前面的数已是不可能),j -= 1;
若ai + aj < sum,ai肯定不是答案之一(前面已得出 j 后面的数已是不可能),i += 1;

###代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution:
def FindNumbersWithSum(self, num, tsum):
# write code here
i = 0
j = len(num) - 1
result = []
while i < j:
if num[i] + num[j] == tsum:
result.append(num[i])
result.append(num[j])
return result
elif num[i] + num[j] > tsum:
j = j - 1
else:
i = i + 1
return result