leetcode 两数之和
思路
给定一个整数数组
nums
和一个目标值target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
class Solution {
public int[] twoSum(int[] nums, int target) {
// 这里写出解
}
}
思路
- 新建一个map,用来存储数据的数据
Map<String, Object> map = new HasMap<>();
- 循环数组将数据中的值作为key,索引作为value,存入map中
for (int i = 0 ; i < nums.length ; i ++) {
map.put(nums[i], i);
}
- 如上图所示,在第一次循环赋值完毕之后, map中的数据如上图右边所示,此时再次循环数据,由题目中给出的目标值
target
与循环数据得到值nums[i]
可以计算得出差值target - nums[i]
,将这个值作为key进入map
中使用map
集合的方法boolean containesKey(Object key)
进行查询, 看是否存在key 如果存在那么久表示,有符合题目描述的两个数字
for(int i = 0 ; i < nums.length ; i ++) {
int cur = target
if(map.containesKey(cur)){
return new int[]{i, map.get(cur)};
}
}
- 注意需要排除题目所说的相同点的情况, 最后整合的结果为
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<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 result = target - nums[i];
if (map.containsKey(result) && map.get(result) != i) {
return new int[]{i, map.get(result)};
}
}
return new int[0];
}
}