给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
拿到这一题,题干中给出假设每种输入只会对应一个答案那就可以使用map,键为元素值,值为元素下标。
起初做法想遍历所有的元素,判断其是否小于target,如果小于就put入map,后来测试发现,该题中数据存在负数,故删去判断。考虑到题中还有[5,5,0],target = 10,这种两个元素的相当的情况,顾加入map.get(flag) !=i
解题代码如下:(时间复杂度O(N)):
public static int[] twoSum(int[] nums, int target){
int[] ints = new int[2];
// 建一个 hashmap存储数据
HashMap<Integer,Integer> map = new HashMap<>();
for (int i=0;i<nums.length;i++){
map.put(nums[i],i);
}
for (int i=0;i<nums.length;i++){
int flag = target - nums[i];
if (map.get(flag) != null && map.get(flag) !=i){
ints = new int[]{i,map.get(flag)};
break;
}
}
return ints;
}