问题
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案,并且你不能使用两次相同的元素。
你可以按任意顺序返回答案。
解题思路
给定一个整数数组 nums
和一个目标值 target
,我们的任务是找到数组中两个整数,使它们的和等于目标值,并返回它们的数组下标。
一个直观的解决办法是双重循环遍历数组,检查所有可能的数对,并找到它们的和是否等于 target
。这种方法的时间复杂度是 O(n²),因为每个元素都要与其他元素进行比较。但我们可以用更高效的方式解决这个问题,即使用 哈希表。
使用哈希表的优化思路
哈希表(HashMap
)允许我们在常数时间内进行查找和插入操作。因此,我们可以利用哈希表来存储数组中的元素和它们的下标,借此减少查找时间。
我们可以通过一次遍历数组来解决问题,具体思路如下
- 对于每个元素
nums[i]
,我们计算目标值target
与当前元素的差值(即complement = target - nums[i]
)。 - 然后检查哈希表中是否存在这个差值
complement
。 - 如果存在,说明之前已经遇到了另一个数字,它与当前数字之和等于目标值
target
,我们可以直接返回它们的下标。 - 如果不存在,则将当前数字和它的下标存入哈希表中,继续处理下一个元素。
代码
class Solution {
public int[] twoSum(int[] nums, int target) {
// 创建一个哈希表来存储数组元素及其索引
Map<Integer, Integer> map = new HashMap<>();
// 遍历数组
for (int i = 0; i < nums.length; i++) {
// 计算目标值与当前元素的差值
int complement = target - nums[i];
// 检查哈希表中是否已经存在差值
if (map.containsKey(complement)) {
// 如果存在,返回当前元素的下标和差值的下标
return new int[] { map.get(complement), i };
}
// 如果不存在,将当前元素和其下标存入哈希表
map.put(nums[i], i);
}
return null;
}
}
复杂度分析
- 时间复杂度:O(n),其中 n 是数组的长度。我们只需遍历一次数组,并且每次查找和插入哈希表的操作都是 O(1) 的。
- 空间复杂度:O(n),最坏情况下哈希表需要存储所有数组中的元素。
总结
利用哈希表,我们可以将两数之和问题从 O(n²) 优化为 O(n),从而大大提高了性能。这种方法特别适用于需要快速查找某些条件的场景,哈希表在这种场景下非常强大。