输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。对应每个测试案例,输出两个数,小的先输出。
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
public class Solution {
public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
ArrayList<ArrayList<Integer>> lists = new ArrayList<>();
int begin = 0;
int end = array.length - 1;
while(begin < end)
{
int currSum = array[begin] + array[end];
if(currSum == sum)
{
ArrayList<Integer> list = new ArrayList<>();
list.add(array[begin]);
list.add(array[end]);
list.add(array[begin] * array[end]);
lists.add(list);
begin++;
end--;
}
else
{
if(currSum < sum)
begin++;
else
end--;
}
}
if(lists.size() ==0) return new ArrayList<Integer>();
//if(lists.size() ==1) return lists.get(0);
Collections.sort(lists, new Comparator<ArrayList<Integer>>() {
@Override
public int compare(ArrayList<Integer> o1, ArrayList<Integer> o2) {
return o1.get(2).compareTo(o2.get(2));
}
});
ArrayList<Integer> result = new ArrayList<>();
ArrayList<Integer> temp = lists.get(0);
for(int i = 0;i < 2; i++){
result.add(temp.get(i));
}
return result;
}
}