题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,输出两个数的乘积最小的。
返回值描述:
对应每个测试案例,输出两个数,小的先输出。
穷举遍历
自己写代码感觉还是很憨批 但是好歹是解出来了
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> FindNumbersWithSum(int [] array,int sum) {
ArrayList<Integer> result = new ArrayList<Integer>();
for (int i = 0; i <array.length ; i++) {
for (int j = i; j < array.length; j++) {
if (array[i]+array[j]==sum){
result.add(array[i]);
result.add(array[j]);
}
}
}
if (result.size()>2){
return hander(result);
}
else {
return result;
}
}
public ArrayList<Integer> hander(ArrayList<Integer> arrayList){
ArrayList<Integer> temp = new ArrayList<Integer>();
for (int i = 0; i < arrayList.size(); i=i+2) {
temp.add(arrayList.get(i)*arrayList.get(i+1));
}
int min = temp.get(0);
for (Integer integer : temp) {
if (integer<min){
min=integer;
}
}
int index = temp.indexOf(min);
ArrayList<Integer> arrayList1 = new ArrayList<Integer>();
arrayList1.add(arrayList.get(index*2));
arrayList1.add(arrayList.get(index*2+1));
return arrayList1;
}
}
更好的解法
双指针,一个从头 一个从尾
因为数组是有序的,所以可以用双指针,指向数组的首尾,具体步骤如下:
1.初始化:指针i指向数组首, 指针j指向数组尾部
2. 如果arr[i] + arr[j] == sum , 说明是可能解
3. 否则如果arr[i] + arr[j] > sum, 说明和太大,所以–j
4. 否则如果arr[i] + arr[j] < sum, 说明和太小,所以++i
但是下面是用c写的
class Solution {
public:
vector<int> FindNumbersWithSum(vector<int> array,int sum) {
if (array.empty()) return vector<int>();
int tmp = INT_MAX;
pair<int, int> ret;
int i = 0, j = array.size();
while (i < j) {
if (array[i] + array[j-1] == sum) {
if (array[i]*array[j-1] < tmp) {
tmp = array[i] * array[j-1];
ret = {i, j-1};
}
++i, --j;
}
else if (array[i] + array[j-1] < sum) {
++i;
}
else {
--j;
}
}
if (ret.first == ret.second) return vector<int>();
return vector<int>({array[ret.first], array[ret.second]});
}
};