【算法题】给定一个排序数字数组和一个目标 K，在数组中找到和为 K 的2个数的坐标。

最新推荐文章于 2025-02-13 16:40:55 发布

原创最新推荐文章于 2025-02-13 16:40:55 发布 · 943 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构

算法题专栏收录该内容

18 篇文章

订阅专栏

该博客讲解了如何使用双指针技巧在已排序数组中寻找两个数之和等于目标值的解决方案。通过左右指针移动调整和的大小，直至找到匹配或指针相遇。代码示例演示了这一过程。

思路

已知输入数组为排序数组，使用两个指针从头和尾对数组进行遍历，

若2数之和大于目标值，则移动右指针，减小当前数的和；

若2数之和小于目标值，则移动左指针，增加当前数的和；

直到找到目标数据，或者2个指针相遇，结束遍历。

代码：

In [1]: def two_index_of_target_sum(array, tar):
   ...:     left, right = 0, len(array)-1
   ...:
   ...:     while left < right:
   ...:         cur_sum = array[left] + array[right]
   ...:         print(f'current sum: {array[left]} + {array[right]} == {cur_sum}')
   ...:         if cur_sum == tar:
   ...:             return left, right
   ...:         elif cur_sum < tar:
   ...:             left += 1
   ...:         else:
   ...:             right -= 1

In [2]: two_index_of_target_sum([1, 2, 3, 4, 6], 6)
current sum: 1 + 6 == 7
current sum: 1 + 4 == 5
current sum: 2 + 4 == 6
Out[2]: (1, 3)