剑指offer57. 和为S的两个数字
解题思路:
利用 HashMap 可以通过遍历数组找到数字组合,时间和空间复杂度均为 O(N) ;
注意本题的 numsnums 是 排序数组 ,因此可使用 双指针法 将空间复杂度降低至 O(1)。
算法流程:
初始化: 双指针 i , j 分别指向数组 nums 的左右两端 (俗称对撞双指针)。
循环搜索: 当双指针相遇时跳出;
计算和 s = nums[i] + nums[j] ;
若 s > target,则指针 j 向左移动,即执行 j = j - 1;
若 s < target ,则指针 i 向右移动,即执行 i = i + 1;
若 s = target ,立即返回数组 [nums[i], nums[j]] ;
返回空数组,代表无和为 target 的数字组合。
# solo 采用对撞指针的方式。如果采用遍历整个列表的方法,时间复杂度为O(n*n)。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
i=0
j=len(nums)-1 # len(nums)-1 才是列表的最后一位数据的索引。
while i<=j: #暂停指标,当对撞指针碰撞时,结束循环。
x=nums[i]+nums[j]
# 判断对撞指针的数据之和和目标数的大小关系
if x==target:
#如果相等就直接返回两个指针指向的数据
return nums[i], nums[j]
elif x<target:
# 如果数据之和小于目标数,说明将前指针i向后移一位。(列表为递增列表,如果是乱序,可以考虑先排序)
i+=1
elif x>target:
# 如果数据之和小于目标数,说明将后指针j向后移一位。
j-=1
"""
提醒一下,判断条件最好不要用相加后的结果,应该用target - nums[i] 跟 nums[j]比较,这样保证不会溢出。虽然这题中不会出错。同样的例子还有二分查找,(left + right) / 2 可以用left + ((rigth - left) >> 1))代替
"""