# 题目描述
给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。
你需要实现的函数 twoSum
需要返回这两个数的下标,并且第一个下标小于第二个下标。注意这里下标的范围是 0 到 n-1。
# 样例
给出 numbers = [2, 7, 11, 15]
, target = 9
, 返回 [0, 1]
.
# 挑战
Either of the following solutions are acceptable:
- O(n) Space, O(nlogn) Time
- O(n) Space, O(n) Time
# 注意事项
你可以假设只有一组答案。
# 问题分析
使用两次循环不难得到两个数的下标,但是若使用一次循环就有点难度了。
我们可以使用 HashMap 结构存储,将元素的值存为 key,下标存为 value,每循环到一个元素时,就判断 map 中是否存在一个与当前元素相加等于 target 的 key,存在我们就可以直接返回这两个元素的下标,不存在我们就将这个元素的值和下标记录在 map 中,以便后续的元素使用。
# 代码实现
public int[] twoSum(int[] numbers, int target) { | |
Map<Integer, Integer> map = new HashMap<>(); | |
int[] index = new int[2]; | |
for (int i = 0; i < numbers.length; i++) { | |
if (map.get(target - numbers[i]) != null) { | |
index[0] = map.get(target - numbers[i]); | |
index[1] = i; | |
return index; | |
} | |
map.put(numbers[i], i); | |
} | |
return index; | |
} |