leetcode#167-两数之和II-输入有序数组

输入一个数组和一个target,找出数组中等于这个target的两个数的下标

题目描述

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

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

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

示例 1:

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

输入:numbers = [2,3,4], target = 6
输出:[1,3]
示例 3:

输入:numbers = [-1,0], target = -1
输出:[1,2]

思路

双指针,一个指针a从头遍历,一个指针b从尾遍历,设 sum = nums[a] + nums[b]

若:

  • sum < target,a++,前面指针往后走,因为后面指针一开始值是最大的(非递减数组)
  • sum > target,b–,后面指针往前走
  • sum == target,可以return [a+1, b+1],就是把下标返回了

复杂度分析:

时间复杂度:O(n),两个指针总遍历次数为数组长度n

空间复杂度:O(1),只使用了两个额外变量

代码

Java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
public int[] twoSum(int[] numbers, int target) {
int a = 0;
int b = numbers.length - 1;
while (a < b) {
if ( numbers[a] + numbers[b] > target) {
b--;
} else if ( numbers[a] + numbers[b] < target){
a++;
} else {
return new int[]{a+1, b+1};
}
}
return new int[]{-1, -1};
}