[开心IT面试题] 在有序数组中查找和正好是给定值的两个数字

题目：

输入一个已经按升序排序过的数组和一个数字，

在数组中查找两个数，使得它们的和正好是输入的那个数字。

要求时间复杂度是O(n)。如果有多对数字的和等于输入的数字，输出任意一对即可。

例如输入数组1、2、4、7、11、15和数字15。由于4+11=15，因此输出4和11。

 

思路：

因为数组是按升序排序过的，所有后面的值一定比前面的值大。

设两个变量elem1和elem2，elem1从数组第一个数开始向后走，elem2从数组最后一个数开始向前走。

求两个变量的和，与给定数字作比较：

当和大于给定数字时，elem2向前走一步，继续比较；

当和小于给定数字时，elem1向后走一步，继续比较； 

当和等于给定数字时，返回当前两个变量值，退出查找。

 

代码：

bool FindSumElements(int array[], int length, int sum, int *elem1, int *elem2)
{
    if(length == 0)
    {
        return false;
    }
    int tempSum = 0;
    //记录elem1和elem2的值分别在数组中下标序号是多少
    //elem1从第一个元素开始，elem2从最后一个元素开始
    int head = 0;
    int tail = length-1;

    while(head < tail)
    {
        tempSum = array[head] + array[tail];
        if(tempSum == sum)
        {
            *elem1 = array[head];
            *elem2 = array[tail];
            return true;
        }else if(tempSum > sum)
        {
            --tail;
        }else
        {
            ++head;
        }
    }

    return false;
}

 

扩展：如果数组是无序的，如何最快查找出这两个数呢？