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; } }

浙公网安备 33010602011771号