题目:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
思路:
- 对于数组中的每个数字nums[i],查找是否存在target-nums[i]
- 用hashmap减少查找的时间消耗
代码:
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 num = target - nums[i];
if (map.containsKey(num) && map.get(num) != i) {
return new int[]{i, map.get(num)};
}
}
throw new IllegalArgumentException("No two sum solution");
}
代码提升:
一次遍历数组就可以得到结果:
假设a + b = target,,那么可以根据a找到b,也可以根据b找到a,那么在进行map赋值的过程中,就可以一次性查找出结果。
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int num = target - nums[i];
if (map.containsKey(num)) {
return new int[] { map.get(num), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}