题目描述:(题目难度,简单)
给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
示例代码
下面给出时间复杂度为,空间复杂度为
的Java代码(可以直接在 leetcode 上提交):
class Solution {
public int[] twoSum(int[] nums, int target) {//前两行是 leetcode 提供的模板
int[] res = {0,0};
HashMap<Integer, Integer> hashMap = new HashMap<>();
for(int i = 0; i < nums.length; ++i){
if(hashMap.size() > 0){
if(hashMap.containsKey(target - nums[i])){
res[0] = hashMap.get(target - nums[i]);
res[1] = i;
return res;
}
}
hashMap.put(nums[i], i);
}
return res;
}
}
思路解析
在遍历数组的过程中,将遍历过的元素都扔到 HashMap 当中。当遍历到当前元素时,查找目标值与当前元素的差值是否在 HashMap 中,如果存在,则算法结束,如果不存在,则将当前元素加入到 HashMap 中,然后遍历下一元素。
为什么能这么做?因为如果输入有解的话,那么要找的那两个元素必定一前一后,当我们遍历到后一元素时,前一元素必定在 HashMap 中,即查找一下 HashMap 便可以得到答案。
网友评论