题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出描述:
对应每个测试案例,输出两个数,小的先输出。
题目地址
https://www.nowcoder.com/practice/390da4f7a00f44bea7c2f3d19491311b?tpId=13&tqId=11195&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
思路
对于一个数组,我们可以定义两个指针,一个从左往右遍历(l),另一个从右往左遍历(r)。首先,我们比较第一个数字和最后一个数字的和curSum与给定数字tsum,
如果curSum == tsum,那么这两个数字就是我们要找的数字,直接输出,这里已经保证了乘积的最小。
如果curSum > tsum,我们需要减小输入值,r向左移动一位。
如果curSum < tsum,我们需要加大输入值,l向右移动一位。
如果没有找到,输出[]。
Python
# -*- coding:utf-8 -*-
class Solution:
def FindNumbersWithSum(self, array, tsum):
# write code here
if len(array)<=1:
return []
left, right = 0, len(array)-1
while left < right:
if array[left]+array[right]==tsum:
return [array[left],array[right]]
elif array[left]+array[right] > tsum:
right -= 1
else:
left += 1
return [] if __name__ == '__main__':
result = Solution().FindNumbersWithSum([1,2,3,4,6],5)
print(result)