其他算法-042-和为S的两个数字(双指针思想)

题目描述

输入一个递增排序的数组和一个数字S，在数组中查找两个数，使得他们的和正好是S，如果有多对数字的和等于S，输出两个数的乘积最小的。

输出描述:

对应每个测试案例，输出两个数，小的先输出。

分析

• 方法一：双层循环，暴力解法，得到所有两数之和，时间复杂度为O($n^2$)。
• 方法二：充分利用数组递增有序的特性，设置两指针到数组的两头，和大了，减小大指针，和小了，增大小指针，相等了返回，两指针相遇退出循环。同理如果输出两个数的乘积最大的，则初始两指针指向中间两个数，往两边移动即可。

代码

# -*- coding:utf-8 -*-
class Solution:
    def FindNumbersWithSum(self, array, tsum):
        # write code here
        
        if not array:
            return []
        
        lowP = 0
        highP = len(array)-1
        
        while lowP<highP:
            if array[lowP]+array[highP]<tsum:
                lowP += 1
            elif array[lowP]+array[highP]>tsum:
                highP -= 1
            else:
                return array[lowP],array[highP]
        
        return []