给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
提示:
2 <= nums.length <= 103
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
1.暴力解法
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] res = new int[2];
int length = nums.length;
for (int i = 0; i < length - 1; i++) {
for (int j = i + 1; j < length; j++) {
if (nums[i] + nums[j] == target) {
res[0] = i;
res[1] = j;
}
}
}
return res;
}
}
2.引入哈希表
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> hashtable = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; ++i) {
if (hashtable.containsKey(target - nums[i])) {
return new int[]{hashtable.get(target - nums[i]), i};
}
hashtable.put(nums[i], i);
}
return new int[0];
}
}
可能会有小白在知道哈希表的检索很快,可以避免遍历后,也无法理解答案。
我简单介绍一下答案的思路,最初创建一个空的哈希表,在遍历数组时,检查哈希表是否包含target - nums[i],如果包含,即找到了要求解的答案,不包含,就要继续遍历,这时要将上一轮遍历到的数据和对应的索引put到哈希表中。
几个可能的疑惑:
最初哈希表为什么是空的,直接把整个数组put进哈希表不行吗?
可以,并且这样做会更容易理解,但会消耗更多内存。代码如下。需要注意的是,不能重复使用同一个索引的元素。
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer, Integer> hashtable = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
hashtable.put(nums[i], i);
}
for (int i = 0; i < nums.length; i++) {
if (hashtable.containsKey(target - nums[i]) && hashtable.get(target - nums[i]) != i) {
return new int[] {hashtable.get(target - nums[i]), i};
}
}
return new int[0];
}
}
为何在遍历完A元素后,才将其put进哈希表?
这样做可以避免重复使用同一个元素。