题目:输入一个递增排序的数组和一个数字s,在数组中找查找两个数,使他们的和正好为s。如果有多对,输出任意一对即可。
def find_nums_sum(nums,s): begin,end=0,len(nums)-1 while begin<end: if 2*nums[begin]<s or 2*nums[end]>s: break if nums[begin]+nums[end]==s: return [nums[begin],nums[end]] elif nums[begin]+nums[end]<s: begin += 1 else: end -= 1 return [-1,-1]
注:由于是有序数组,采用前后两个指针往中间走的方式,如果两个数的和小于目标,前指针后移,反之,后指针前移。代码中增加了优化,如果目标值大于两倍的后指针或小于两倍的前指针则数组中不会存在这两个数,提前终止。