LeetCode:167. 两数之和 II - 输入有序数组

给定一个已按照 升序排列  的整数数组 numbers ,请你从数组中找出两个数满足相加之和等于目标数 target 。

函数应该以长度为 2 的整数数组的形式返回这两个数的下标值。numbers 的下标 从 1 开始计数 ,所以答案数组应当满足 1 <= answer[0] < answer[1] <= numbers.length 。

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

示例 :

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

相比于LeetCode第1题的两数之和,这个题目在于数组是有序的。

因此很快能想到二分查找。

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

时间复杂度O(nlog(n)):全部遍历一次O(n) * 每个遍历时二分法查找一次O(log(n))。

空间复杂度:O(1)。

那么,还有没有更好的方案?

显然,双指针又派上用场了,两个指针分别从头尾开始,f(0++)+f((n-1)--),只要这个数大于target,那么右指针往前走,反之,左指针往后走,直到找到等于target的两个数。

class Solution {
    public int[] twoSum(int[] numbers, int target) {
        int left = 0,right= numbers.length-1;

        while(left<right){

            if(numbers[left]+numbers[right]>target){
                right--;
            } else if(numbers[left]+numbers[right]<target){
                left++;
            }else {
                return new int[]{left+1,right+1};
            }

        }
        return null;
    }
}

 

posted @ 2021-04-22 14:33  蜗壳吃虾米  阅读(45)  评论(0)    收藏  举报