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

0 阅读1分钟

题目

给你一个下标从 1 开始的整数数组 numbers ,该数组已按 ****非递减顺序排列 ** ,请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] 和 numbers[index2] ,则 1 <= index1 < index2 <= numbers.length 。

以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 **和 **index2

你可以假设每个输入 只对应唯一的答案 ,而且你 不可以 重复使用相同的元素。

你所设计的解决方案必须只使用常量级的额外空间。

示例 1:

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

题解

原始的两数之和采用哈希表,空间不是常量级

方式一:双指针

public int[] twoSum(int[] numbers, int target) {
    int[] result = new int[2];

    int l = 0;
    int r = numbers.length - 1; 
    // 不能重复使用 所以不是 l <= r
    while (l < r) {
        int sum = numbers[l] + numbers[r];
        if (sum > target) {
            r--;
        } else if (sum < target) {
            l++;
        } else {
            result[0] = l + 1;
            result[1] = r + 1;
            return result;
        }
    }

    return result;
}

方式二:二分

public int[] twoSum(int[] numbers, int target) {
    for (int i = 0; i < numbers.length; i++) {
        // i 已确定,在[i + 1, length - 1]中找另一个下标
        int low = i + 1;
        int high = numbers.length - 1;
        while (low <= high) {
            int mid = low + (high - low) / 2;
            if (numbers[mid] == target - numbers[i]) {
                return new int[]{i + 1, mid + 1};
            } else if (numbers[mid] > target - numbers[i]) {
                high = mid - 1;
            } else {
                low = mid + 1;
            }
        }
    }
    return new int[2];
}

总结

算法:双指针二分