输入一个已经按升序排序过的数组和一个数字,在数组中查找两个数,使得它们的和正好是输入的那个数字。要求时间复杂度是O(n)。如果有多对数字的和等于输入的数字,输出任意一对即可。
例如输入数组1、2、4、7、11、15和数字15。由于4+11=15,因此输出4和11。
设和为 sum,数组为a
分析:
(1)遍历一遍数组,当前遍历到a[i]。由于生序排列,可以二分查询sum-a[i]是否在a中的复杂度为O(lgn),整个算法复杂度为O(nlgn),不合题意。
(2)对于每一个数a[i],关键问题是查询sum-a[i]是否在a中。题目中升序排列这个条件还有一个性质,
对于j>i --> a[j] > a[i] --> sum-a[i] < sum-a[j]
这说明了,我们在查询完sum-a[i]是否在数组中后(假设a[j]<sum-a[i], a[j+1]>=sum-a[j]),查询sum-a[i+1]可以缩小到[i+1, j]这个范围。
#include <iostream>
using namespace std;
bool find(int* a, int len, int sum, int& x) {
int right = len-1;
for(int i = 0; i <= right; i++) {
x = sum-a[i];
while(right >= 0 && a[right] > x) right --;
if(right == -1) break;
if(a[right] == x) {
return true;
}
}
return false;
}
int main() {
int a[6] = {1, 2, 4, 7, 11, 15};
int x, sum = 15;
if(find(a, 6, sum, x)) {
cout << sum-x << " " << x << endl;
}
return 0;
}