题目
给定一个整数数组
nums
和一个整数目标值target
,请你在该数组中找出和为目标值target
的那两个整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。
解题
可以使用哈希表(字典)来高效地解决这个问题。这种方法的时间复杂度是
,因为只需要遍历一次数组。
具体步骤:
- 创建一个空的哈希表,用于储存数组中的元素及其对应的下标
- 遍历数组,对于每个元素 nums[i]:
- 计算 target 与当前元素 nums[i] 的差值。
- 检查哈希表中是否存在这个差值:存在则说明找到了并返回下标;不存在则将当前元素 nums[i] 及其下标存入哈希表
def two_sum(nums, target):
# 创建一个哈希表来存储元素及其下标
num_to_index = {} # 字典
# 遍历数组
for i, num in enumerate(nums):
# 计算差值
diff = target - num # 每次遍历时计算
# 检查差值是否在哈希表中
if diff in num_to_index:
return [num_to_index[diff], i]
# 将当前元素及其下标存入哈希表
num_to_index[num] = i # 不在哈希表中
# 如果没有找到答案,返回空列表
return []
if __name__ == '__main__':
nums = [7, 2, 11, 15]
target = 9
print(two_sum(nums, target))
[0, 1]
总结
哈希表的基本原理和应用
哈希表是一种数据结构,它能够以接近常数时间
进行插入、删除和查找操作。在这个题目中,哈希表用来存储数组中的每个元素及其对应的下标,从而可以在常数时间内查找是否存在某个数值。
一次遍历的技巧
该题目要求找到两个数的和为目标值,最直接的方式是双重循环,时间复杂度为
。使用哈希表可以将时间复杂度优化为
,这是通过一次遍历解决问题的关键。
查找与插入的顺序
在遍历数组时,先查找再插入元素到哈希表中是关键。如果先插入再查找,可能会错误地找到同一个元素作为答案。例如,输入
[3, 3]
和目标值6
时,如果先插入再查找,会错误地找到同一个元素。
数组与字典的结合
需要理解如何将数组元素与其索引关联存储在字典中,并利用字典的高效查找特性快速找到所需的两个数的索引。