题目一:
输入一个递增排序的数组和一个数字s,在数组中查找两个数,使得它们的和正好是s。如果有多对数字的和等于s,输出任意一对即可。要求O(n)。
分析:
增加两个指针,分别指向数组的第一个和最后一个元素,如果指针指向的两个元素之和大于s,后面的指针向前移动,如果指针指向的两个元素之和小于s,则前面的指针向后移动。只需要遍历数组一次,所以时间复杂度为O(n)。
class Solution
{
public:
bool FindNumbersWithSum( vector<int> data, int sum, int *num1, int *num2 )
{
int i = 0, j = data.size()-1;
if (num1 == NULL || num2 == NULL)
return false;
while ( i < j )
{
int tempSum = data[i] + data[j];
if ( tempSum == sum )
{
*num1 = data[i];
*num2 = data[j];
return true;
}
else if ( tempSum < sum )
{
i++;
}
else
{
j--;
}
}
return false;
}
};
int main( void )
{
vector<int> data = { 1, 2, 4, 7, 11, 15 };
//vector<int> data = {};
int num1, num2;
Solution sos;
cout << sos.FindNumbersWithSum( data, 15, &num1, &num2 ) << endl;
cout << num1 << " " << num2 << endl;
return 0;
}
题目二:
输入一个正数s,打印出所有和为s的连续正数序列(至少含有两个数)。例如输入15,由于1+2+3+4+5=4+5+6=7+8=15,所以结果打印出3个连续序列1~5、4~6和7~8。
分析:
增加两个指针,这两个指针指向确定数组中的一个区间,如果这个区间中的数字之和大于s则减小区间大小,否则增加区间大小。
class Solution
{
public:
vector< vector<int> > FindContinuousSequence( int sum )
{
if ( sum < 3 )
return vector< vector<int> >();
vector< vector<int> > result;
int start = 1, end = 2;
int curSum = start+end;
while ( start <= sum/2 && end <= sum )
{
if ( curSum == sum )
{
vector<int> temp;
for ( int i = start; i <= end; i++ )
{
temp.push_back( i );
}
result.push_back( temp );
}
while ( curSum > sum && start <= sum/2 && end <= sum )
{
curSum -= start;
start++;
if ( curSum == sum )
{
vector<int> temp;
for ( int i = start; i <= end; i++ )
{
temp.push_back( i );
}
result.push_back( temp );
}
}
end++;
curSum += end;
}
return result;
}
};
int main( void )
{
Solution sos;
vector< vector<int> > result;
result = sos.FindContinuousSequence( 15 );
for ( int i = 0; i < result.size(); i++ )
{
for ( int j = 0; j < result[i].size(); j++ )
{
cout << result[i][j] << " ";
}
cout << endl;
}
return 0;
}