输入一个数组和一个数字,在数组中查找两个数,使得它们的和正好是输入的那个数字。
要求时间复杂度是O(N)。如果有多对数字的和等于输入的数字,输出任意一对即可。
例如输入数组1、2、4、7、11、15和数字15。由于4+11=15,因此输出4和11。
#include <iostream>
using namespace std;
#define SIZE 10
void initRandom(int* arr) {
srand(unsigned(time(0)));
for (int i = 0; i < SIZE; i++) {
arr[i] = rand() % 20;
}
}
int myCmp(const void *a, const void *b) {
return *(int*) a - *(int*) b;
}
int find2NumEqToSum(int* arr, int sum, int &a, int &b, int &index1,
int &index2) {
int flag = 0;
int begin = 0;
int end = SIZE - 1;
while (begin < end) {
if (arr[begin] + arr[end] < sum) {
begin++;
} else if (arr[begin] + arr[end] > sum) {
end--;
} else {
flag = 1;
if (a != arr[begin]) {
a = arr[begin];
b = arr[end];
index1 = begin;
index2 = end;
cout << "a: " << a << " index1:" << index1 << " , b: " << b
<< " index2:" << index2 << endl;
}
begin++;
end--;
// return 1;
}
}
return flag;
}