Bootstrap

python 利用双指针法(左右指针)求取有序数组中的两数之和

利用双指针法(左右指针)求取有序数组中的两数之和

左右指针(也称为对撞指针):

主要用于在有序数组中查找两数之和、反转数组等问题。
一个指针从数组的起始位置开始,另一个指针从数组的末尾开始,两个指针向中间移动。


from typing import Union, List

def tow_sum(_nums: List[Union[int, float]], _target: Union[int, float], _reverse: bool = False):
    """

    Args:
        _nums: 有序数组
        _target: 两数之和的目标值
        _reverse: true代表_nums是降序,false代表是升序

    Returns:
        找到以元组的形式返回两个数的索引
        没找到返回None
    """
    _left, _right = 0, len(_nums) - 1

    while _left < _right:
        _sum = _nums[_left] + _nums[_right]

        if _sum == _target:
            return _left, _right
        elif _sum > _target:
            if _reverse:
                _left += 1
            else:
                _right -= 1
        else:
            if _reverse:
                _left -= 1
            else:
                _right += 1
    return None


if __name__ == '__main__':
    # 一个升序数组
    test_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    print(tow_sum(_nums=test_list, _target=10))  # (0, 8)
    test_list.reverse()
    print(tow_sum(_nums=test_list, _target=10, _reverse=True))  # (1, 9)


;