解题思路:
- 创建一个空字典,用于存储数组中每个元素的索引。
- 遍历数组,对于每个元素:
- 计算目标值与当前元素的差值
diff
。 - 检查字典中是否存在
diff
,如果存在,则找到了满足条件的两个数。 - 如果不存在,则将当前元素及其索引添加到字典中。
- 如果没有找到满足条件的两个数,则返回一个空列表。
- 如果找到了满足条件的两个数,则返回它们的索引。
def two_sum(nums, target):
# 创建一个空字典,用于存储数组中每个元素的索引
num_dict = {}
# 遍历数组
for i, num in enumerate(nums):
# 计算目标值与当前元素的差值
diff = target - num
# 检查字典中是否存在差值
if diff in num_dict:
# 找到了满足条件的两个数,返回它们的索引
return [num_dict[diff], i]
# 将当前元素及其索引添加到字典中
num_dict[num] = i
# 没有找到满足条件的两个数,返回一个空列表
return []
# 测试
nums = [2, 7, 11, 15]
target = 9
result = two_sum(nums, target)
print(result) # 输出: [0, 1],因为 nums[0] + nums[1] = 2 + 7 = 9
该算法的时间复杂度为 O(n),其中 n 是数组的长度。在算法的执行过程中,我们只需要遍历一次数组,对于每个元素,可以在常数时间内进行查找操作。因此,该算法的时间复杂度是线性的。