【leetcode】167. 两数之和 II - 输入有序数组（two-sum-ii-input-array-is-sorted）（二分）[简单]

链接

https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted/

耗时

解题：16 min
题解：4 min

题意

给定一个已按照升序排列 的有序数组，找到两个数使得它们相加之和等于目标数。

函数应该返回这两个下标值 index1 和 index2，其中 index1 必须小于 index2。

说明:

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

思路

因为数组有序，遍历数组，二分找 目标数-当前元素 的值，如果能找到，当前元素位置和找到的位置即为答案。

时间复杂度：$O(nlogn)$

AC代码

class Solution {
public:
    vector<int> twoSum(vector<int>& numbers, int target) {
        int n = numbers.size();
        vector<int> ans;
        for(int i = 0; i < n; ++i) {
            int index2 = lower_bound(numbers.begin(), numbers.end(), target-numbers[i]) - numbers.begin();
            if(index2 < n && numbers[i]+numbers[index2] == target) {
                if(index2 == i) {
                    if(index2+1 < n && numbers[index2+1] == numbers[i]) {
                        index2 += 1;
                    }
                    else {
                        continue;
                    }
                }
                ans.push_back(i+1);
                ans.push_back(index2+1);
                break;
            }
        }
        return ans;
    }
};