求解思路
一、暴力算法
二、使用集合降低复杂度
整体思路
遍历数组,用 Target 减去当前元素,判断当前计算结果是否包含在剩余的元素中。
细节实现
- 判断使用的数据结构:上述思路中设计判断一个元素是否存在,集合可以在O(1)的时间内完成该操作,因此本题目考虑使用集合。
- 选择集合的类型:当判断差值存在于剩余的元素中时,需要返回该元素的下标值,所以这里需要一个键值对建立实际值与下标值之间的关系。因此选取实现了Set接口的方法,由于这里的键值对对于顺序没有要求,所以这里选取HashMap存储数据。
- 计算得到的差值先判断是否存在于集合中,若存在。则取出集合中的该元素与当前遍历的数组元素作为结果进行返回;若不存在,则将该键值对加入到集合中,集合中始终存放的是遍历完的元素。
代码实现
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<>();
int[] res = new int[2];
for (int i = 0; i < nums.length; i++) {
int temp = target - nums[i];
if(map.containsKey(temp)){
res[1] = i; //因为假设只有一种结果,所以这里不需要跳出
res[0] = map.get(temp);
}
map.put(nums[i],i); // 存放遍历过的<下标,值>
}
return res;
}
注意
- 泛型在使用时进行指定的写法是更加规范的,即使在默认情况下是Object
Map<Object,Object> map = new HashMap<>(); // <Object,Object>不写也行,但是标注更加规范
- HashMap为了方便遍历,将key的值存放在实现Set接口的类中,将value的值存放在实现Collection接口的类中