两数之和
原题
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 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 <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
- 只会存在一个有效答案
class Solution {
public int[] twoSum(int[] nums, int target) {
}
}
解题思路
- 暴力解法:通过双重循环遍历数组中的所有可能的数字组合来寻找满足条件的两个数。
- 优化解法。
代码示例
class Solution {
public int[] twoSum(int[] nums, int target) {
int length = nums.length;
for (int i = 0; i < length; i++) {
for (int j = i + 1; j < length; j++) {
if (nums[i] + nums[j] == target) return new int[]{i, j};
}
}
return new int[0];
}
}
优化
在这道题中,当我们遍历数组时,希望快速找到是否存在一个已经遍历过的数,使得该数和当前数的和等于目标值。而哈希表提供接近 O(1)
时间复杂度的查找性能,这使得它非常适合用来在一组数据中快速找到所需元素。
使用哈希表避免了双重 for 循环中不必要的重复遍历,显著降低了时间复杂度。虽然需要额外的存储空间来保存已经遍历过的元素,但使这部分空间消耗是可以接受的,特别是对于时间性能要求高的场景。
思路
- 初始化一个哈希表来存储数组元素及其对应的索引。
- 遍历数组中的每个元素,计算目标值与当前元素之间的差值。
- 如果哈希表中包含这个差值,说明找到了两个数的和等于目标值,返回当前元素的索引和差值的索引。
- 如果哈希表中不包含这个差值,将当前元素及其索引添加到哈希表中。
- 找到符合条件的两个数,返回它们的索引(/遍历结束仍未找到,返回空数组)。
举例
nums = [2,7,11,15]
target = 18
初始 HashMap
map
为空第一次迭代,当前元素为
2
,差值为18 - 2 = 16
,map
中不包含16
,将当前元素及其索引添加到哈希表中,哈希表变为{2: 0}
。第二次迭代,当前元素为
7
,差值为18 - 7 = 11
,map
中不包含11
,将当前元素及其索引添加到哈希表中,哈希表变为{2: 0, 7: 1}
。第三次迭代,当前元素为
11
,差值为18 - 11 = 7
,map
中包含7
,找到结果,返回7
对应的索引值和当前元素的索引,即1
和2
。
优化后代码
class Solution {
public int[] twoSum(int[] nums, int target) {
// 创建哈希表用于存储数组元素的值和该元素在数组中的索引
HashMap<Integer, Integer> map = new HashMap<>();
// 遍历数组直到找到结果
for (int i = 0; i < nums.length; i++) {
// 如果差值在哈希表中则返回数组索引表明找到结果
if (map.containsKey(target - nums[i])) {
return new int[]{map.get(target - nums[i]), i};
}
// 如果差值不在哈希表中则添加该元素至哈希表
map.put(nums[i], i);
}
// (未找到返回空数组)
return new int[0];
}
}