剑指42 - 和为S的两个数字
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
输出描述:
对应每个测试案例,输出两个数,小的先输出。
示例1
输入
[1,2,4,7,11,15],15
输出
[4,11]
解法一:
import java.util.*;
public class Solution {
public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
ArrayList<Integer> result = new ArrayList<>();
if(array == null || array.length <2) return result;
//一个Map存放 k=乘积,v=list
Map<Integer, ArrayList<Integer>> retMap = new HashMap<>();
int pHi = array.length-1, pLo=0;
while(pHi > pLo) {
int tempSum = array[pHi] + array[pLo];
if(tempSum == sum) {
ArrayList<Integer> tempList = new ArrayList<>();
tempList.add(array[pLo]);
tempList.add(array[pHi]);
retMap.put(array[pHi]*array[pLo], tempList);
pHi--;
pLo++;
}else if(tempSum < sum) {
// 扩大窗口,右指针右移
pLo++;
}else {
pHi--;
}
}
int minKey = Integer.MAX_VALUE;
for(Integer key: retMap.keySet()){
if(key < minKey) minKey = key;
}
return retMap.get(minKey)==null?result:retMap.get(minKey);
}
}
优化:思路 -> 头尾两指针向中间夹击的时候,第一次拿到的两个数,乘积就是最小的!
import java.util.*;
public class Solution {
public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
ArrayList<Integer> result = new ArrayList<>();
if(array == null || array.length <2) return result;
int pHi = array.length-1, pLo=0;
while(pHi > pLo) {
int tempSum = array[pHi] + array[pLo];
if(tempSum == sum) {
result.add(array[pLo]);
result.add(array[pHi]);
return result;
}else if(tempSum < sum) {
// 扩大窗口,右指针右移
pLo++;
}else {
pHi--;
}
}
return result;
}
}