BM50-两数之和
题目
给出一个整型数组 numbers 和一个目标值 target,请在数组中找出两个加起来等于目标值的数的下标,返回的下标按升序排列。
(注:返回的数组下标从1开始算起,保证target一定可以由数组里面2个数字相加得到)
数据范围:2≤len(numbers)≤10^5,−10≤numbersi≤10^9,0≤target≤10^9。
要求:空间复杂度 O(n),时间复杂度 O(nlogn)。
示例1
输入:[3,2,4],6
返回值:[2,3]
说明:因为 2+4=6 ,而 2的下标为2 , 4的下标为3 ,又因为 下标2 < 下标3 ,所以返回[2,3]。
示例2
输入:[20,70,110,150],90
返回值:[1,2]
说明:20+70=90。
思路
使用Map来降低时间复杂度,遍历数组,如果没有 (target - 当前值) 就将当前数字存入哈希表,如果有,返回该数字下标即可。
哈希表可以优化第二遍循环中对元素的检索速度。
代码
import java.util.*;
public class Solution {
/**
*
* @param numbers int整型一维数组
* @param target int整型
* @return int整型一维数组
*/
public int[] twoSum (int[] numbers, int target) {
HashMap<Integer, Integer> map = new HashMap<>();
//遍历数组
for(int i = 0; i < numbers.length; i++) {
//将不包含target - numbers[i]装入map中,包含的话直接返回下标
if(map.containsKey(target - numbers[i])) {
return new int[]{map.get(target - numbers[i]) + 1, i + 1};
} else {
map.put(numbers[i], i);
}
}
throw new IllegalArgumentException("No solution");
}
}
- 时间复杂度:O(n) 一次遍历hash索引查找时间复杂度为O(1)。
- 空间复杂度:O(n) 申请了n大小的map空间。