Leetcode167. 两数之和

2020-01-06  本文已影响0人  LonnieQ

题目

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

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

说明:
返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。

解题思路

遍历数组的元素,并用二分搜索尝试寻找和当前数相加为目标值的元素。同时主要要跳过重复选项。

C++解法

class Solution {
public:
    int binarySearch(vector<int> numbers, int from, int to, int target) {
        if (from > to) return -1;
        int mid = (from + to) / 2;
        if (target == numbers[mid]) return mid;
        else if (target < numbers[mid]) return binarySearch(numbers, from, mid - 1, target);
        else return binarySearch(numbers, mid + 1, to, target);
    }
    vector<int> twoSum(vector<int>& numbers, int target) {
        int result = 0;
        int n = (int)numbers.size();
        for (int i = 0; i < numbers.size(); ++i) {
            if (i > 0 && numbers[i] == numbers[i - 1]) continue;
            if (numbers[i] > target) break;
            if ((result = binarySearch(numbers, i + 1, n - 1, target - numbers[i])) != -1) {
                return {i + 1, result + 1};
            }
        }
        return {};
    }
};

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted

上一篇下一篇

猜你喜欢

热点阅读