问题描述
问题分析
考虑到整数数组中每个元素的值与它的下标要相互对应,想到了可以使用Hash表结构(这里使用HashMap)。使用Hash表的策略是利用空间换时间,将查找每个元素的时间降低到了O(1)。我们可以使用一次迭代来完成整个过程。迭代过程中,查找HashMap中与该元素互补的元素(key)的下标(value),如果找到就返回当前元素和互补元素的下标;如果没有找到,就将该元素加入到HashMap中。
- 时间复杂度:O(n), 我们只遍历了包含有n个元素的列表一次。在表中进行的每次查找只花费O(1)的时间。
- 空间复杂度:O(n), 所需的额外空间取决于哈希表中存储的元素数量,该表最多需要存储n个元素。
Java代码
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap map = new HashMap();
for(int i=0; i<nums.length; i++){
Integer index = (Integer)map.get(target - nums[i]);
if(index != null){
return new int[]{index,i};
}else{
map.put(nums[i],i);
}
}
return new int[]{};
}
}