和为S的两个数字

最新推荐文章于 2021-11-07 16:02:29 发布

原创最新推荐文章于 2021-11-07 16:02:29 发布 · 139 阅读

0 ·

CC 4.0 BY-SA版权

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

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

定义两个索引，初始化索引begin为0，即数组array的第一个值；初始化索引end为length-1，即为数组array的最后一个值。
只要begin小于end，遍历数组。
当二数相加等于S时，返回这两个数，因为两数相乘，二数差值越大，乘积越小，所以找到的第一组满足条件的数字，即为结果。
当二数相加小于S时，begin+1。
当二数相加大于S时，end-1。

python代码：

# -*- coding:utf-8 -*-
class Solution:
    def FindNumbersWithSum(self, array, tsum):
        # write code here
        length = len(array)
        if not array or length == 0:
            return []

        begin = 0
        end = length - 1
        result = []
        while begin < end:
            currentSum = array[begin] + array[end]
            if currentSum == tsum:
                return [array[begin], array[end]]
            elif currentSum < tsum:
                begin += 1
            else:
                end -= 1

        return []

if __name__ == '__main__':
    solution = Solution()
    print(solution.FindNumbersWithSum([1,2,3,4,5],6))