2020/12/1 二分法练习

以167. 两数之和 II - 输入有序数组为例

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

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

说明:

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

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

首先给出最原始的解法时间复杂度为O(n2)

class Solution {
    public int[] twoSum(int[] numbers, int target) {
        int tmp;
        //双重遍历求得
        for(int i=0;i<numbers.length;i++){
            tmp = target - numbers[i];
            for(int j=i+1;j<numbers.length;j++){
                if(tmp == numbers[j]){
                    int[] a = new int[] {i+1,j+1};
                    return a;
                }
            }
        }
        return null;
    }
}

由于数组是有序数组，在第二次遍历时，可以借助二分法降低时间复杂度至O(nlogn)，此处二分法使用时，由于可以直接从第二个元素进行遍历，因此left可以加1，相应的需要对right进行减1操作。

class Solution {
    public int[] twoSum(int[] numbers, int target) {
        int tmp;
        for(int i=0;i<numbers.length;i++){
            int left = i+1;
            int right = numbers.length-1;
            tmp = target - numbers[i];
            while(left <= right){
                int mid = (left + right)/2;
                if(tmp == numbers[mid]){
                    return new int[] {i+1, mid+1};
                }
                else if(tmp < numbers[mid]){
                    right = mid-1;
                }
                else if(tmp > numbers[mid]){
                    left = mid+1;
                }
            }
        }
        return null;
    }
}

方法三：双指针法
可以通过两个指针分别从两端向中间遍历，时间复杂度为O(n)。

class Solution {
    public int[] twoSum(int[] numbers, int target) {
        int left = 0;
        int right = numbers.length-1;
        while(left < right){
            if(numbers[left] + numbers[right] == target){
                return new int[]{left+1,right+1};
            }
            else if(numbers[left] + numbers[right] > target){
                right--;
            }
            else{
                left++;
            }
        }
        return null;
    }
}