版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/AnselLyy/article/details/80953734
1. Two Sum
欢迎访问 我的个人博客
Description
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
Example
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
Thinking
- 返回数组 nums 中两个值的下标,这两个值的和等于所给的 target
- 数组 nums 中的元素并不是有序的
- 假定解是唯一的,并且每个元素不得重复使用
- 需要特别注意元素是 target 的一半的情况
- 还需要特别注意一个值重复出现的情况
Solution -1
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
// 保存每个值到 map 中,value 取用同 target 的差
for (int i = 0; i < nums.length; i++) {
if ((target-nums[i]) == nums[i] && !map.containsKey(nums[i])) {
map.put(nums[i], -1);
} else {
map.put(nums[i], target-nums[i]);
}
}
int[] result = new int[2];
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(map.get(nums[i]))) {
result[0] = i;
break;
}
}
for (int i = 0; i < nums.length; i++) {
if (i != result[0] && nums[i] == map.get(nums[result[0]])) {
result[1] = i;
break;
}
}
return result;
}
Solution -2
大神做法,借鉴学习
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> res = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int difference = target - nums[i];
if (res.containsKey(difference)) {
return new int[]{ res.get(difference), i };
}
res.put(nums[i], i);
}
return null;
}