題目:輸入一個已經按升序排序過的數組和一個數字,
在數組中查找兩個數,使得它們的和正好是輸入的那個數字。
要求時間複雜度是O(n)。如果有多對數字的和等於輸入的數字,輸出任意一對即可。
例如輸入數組1、2、4、7、11、15和數字15。由於4+11=15,因此輸出4和11。
思路:由於數組爲升序,因而可以設置頭尾兩個指針,從第一個元素和最後一個元素相加,如果大於sum, 則尾指針向前移1個元素,繼續判斷兩個數的和;如果小於sum,則頭指針向後移1個元素,繼續判斷兩個數的和。
#include <iostream>
#include <stdio.h>
#include <cstdlib>
using namespace std;
bool search(int A[], int &sum, int &begin, int &end)
{
bool find = false;
while(begin < end)
{
if(A[begin]+A[end] > sum)
end--;
else if(A[begin]+A[end] < sum)
begin++;
else
{
find = true;
break;
}
}
return find;
}
int main()
{
int begin = 0, end = 5, sum = 15;
int A[100] = {1, 2, 4, 7, 11, 15};
if(search(A, sum, begin, end))
cout << A[begin] << " " << A[end] << endl;
return 0;
}