剑指Offer 42. 和为S的两个数字 (其他)

Posted on 2018-10-16 12:40  _hqc  阅读(143)  评论(0编辑  收藏  举报

题目描述

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

输出描述:

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

题目地址
https://www.nowcoder.com/practice/390da4f7a00f44bea7c2f3d19491311b?tpId=13&tqId=11195&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

思路

对于一个数组,我们可以定义两个指针,一个从左往右遍历(l),另一个从右往左遍历(r)。首先,我们比较第一个数字和最后一个数字的和curSum与给定数字tsum,

如果curSum == tsum,那么这两个数字就是我们要找的数字,直接输出,这里已经保证了乘积的最小。

如果curSum > tsum,我们需要减小输入值,r向左移动一位。

如果curSum < tsum,我们需要加大输入值,l向右移动一位。

如果没有找到,输出[]。


Python

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

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