剑指Offer:和为S的两个数字Java/Python

1.题目描述

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

2.算法描述

方法:双指针

1. l o w , h i g h \red{声明两个指针low,high分别指向数组的开始和结尾。}
2. a r r a y [ l o w ] + a r r a y [ h i g h ] s u m \red{比较array[low]+array[high]与sum:}
如果 a r r a y [ l o w ] + a r r a y [ h i g h ] = = s u m \red{array[low]+array[high] == sum} :找到满足要求的两个数。
如果 a r r a y [ l o w ] + a r r a y [ h i g h ] < s u m \red{array[low]+array[high] < sum} :和不够sum,需要low自增1(数组是递增的)
如果 a r r a y [ l o w ] + a r r a y [ h i g h ] > s u m \red{array[low]+array[high] > sum} :和超出sum,需要high自减1(数组是递增的)

为什么这个时候一定是积最小的?

高中的基本不等式可以解答:简单来说,就是当两个正数 x y x,y的和 为定值的时候, x y x*y 的最大值是 x = y x=y 的时候。基本不等式

3.代码描述

3.1.Java代码

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
        ArrayList<Integer> ans = new ArrayList<>();
        int l = 0, r = array.length-1;
        while(l < r){
            if(array[l] + array[r] == sum){
                ans.add(array[l]);
                ans.add(array[r]);
                break;
            }
            else if(array[l] + array[r] > sum){
                r--;
            }
            else if(array[l] + array[r] < sum){
                l++;
            }
        }
        return ans;
    }
}

3.2.Python代码

# -*- coding:utf-8 -*-
class Solution:
    def FindNumbersWithSum(self, array, tsum):
        low = 0
        high = len(array)-1
        ans = []
        while low<high:
            if array[low] + array[high] == tsum:
                ans.append(array[low])
                ans.append(array[high])
                break
            elif array[low] + array[high]<tsum:
                low += 1
            else:
                high -=1
        return ans

猜你喜欢

转载自blog.csdn.net/qq_37888254/article/details/90201048