【Leetcode】167. 两数之和II - 输入有序数组

QUESTION

easy

题目描述

给定一个已按照升序排列的有序数组，找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2，其中 index1 必须小于 index2。

示例:

输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。

说明

• 返回的下标值（index1 和 index2）不是从零开始的。
• 你可以假设每个输入只对应唯一的答案，而且你不可以重复使用相同的元素。

---

SOLUTION

方法一

一开始就想到了二分查找，根据题目的意思，每次输入都只有一个唯一的答案，所以我们按照，定左边，找右边的方式完成整个程序的编写即可，O(nlogn)。

class Solution {
public:
    vector<int> twoSum(vector<int>& numbers, int target) {
        for (int i = 0; numbers[i] <= target; ++i) {
            int t = target - numbers[i], l = i + 1, r = numbers.size() - 1;
            while (l <= r) {
                int mid = l + (r - l) / 2;
                if (numbers[mid] < t) l = mid + 1;
                else if(numbers[mid] > t) r = mid - 1;
                else return {i + 1, mid + 1};
            }
        }
        return {};
    }
};

方法二

方法二采用头尾两个指针，逐步将范围缩小。事实上方法一在某些情况下可以比方法二更高效，特别是在优化一下搜索的情况后。但是按时间复杂度来说，方法二是要更胜一筹的，为 O(n)。

class Solution {
public:
    vector<int> twoSum(vector<int>& numbers, int target) {
        int index1 = 0, index2 = numbers.size() - 1;
        while(index1 < index2){
            if(numbers[index1] + numbers[index2] > target) index2--;
            else if(numbers[index1] + numbers[index2] < target) index1++;
            else return {index1 + 1, index2 + 1};
        }
        return {};
    }
};