给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
LeetCode 1. 两数之和
菜鸡最容易想到的方法就是暴力法了,两个for循环枚举每个可能的组合,写着试试
class Solution {
public int[] twoSum(int[] nums, int target) {
int res[]=new int[2];
for(int i=0;i<nums.length;i++){
for(int j=i+1;j<nums.length;j++){
if(nums[i]+nums[j]==target){
res[i]=i;
res[j]=j;
return res;
}
}
}
return res;
}
}
时间复杂度为O(n的平方)
看继续优化:
每次从数组中取出一个值,并且判断target-nums[i]是否在HashMap中,在的话,说明已经找到了。没在的话,就将这个值加入HashMap中。
这里需要注意的是将num[i]作为key,而不是将i作为key。
因为我们每次判断target-nums[i]是否在HashMap中时,用到map.containsKey这个方法,将num[i]作为key的话,可以根据HashCode直接查找到是否包含。而如果将i作为key的话,则是需要用到map.containsValue这个方法,HashMap执行这个方法时,是需要将HashMap全部遍历一遍,找到value值;时间又大大增加了。
这里的key不可能重复,因为题目要求的只有一个解,如果返回结果的第一个元素有两个值的话,则不满足题目。所以就算发生了哈希冲突,这里冲突的这个值也不可能是答案中的一个值
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> map=new HashMap<>();
for(int i=0;i<nums.length;i++){
//包含总和为target的值
if(map.containsKey(target-nums[i])){
return new int[]{i,map.get(target-nums[i])};
}
//不存在的话,将其加入到map中
map.put(nums[i],i);
}
return null;
}
}
时间复杂度为O(1),只需要遍历nums数组即可。从HashMap中判断是否有值时,时间复杂度为O(1)
空间复杂度为O(n)
总结:有时候思考题目时,可以尝试多方面思考,不要总是一个方向走到死