原始经典解法不再赘述,使用哈希表可以将时间复杂度降低到 O(n)
from typing import List
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
num_map = {}
for i, num in enumerate(nums):
complement = target - num
if complement in num_map:
return [num_map[complement], i]
num_map[num] = i
return [] # 如果没有找到符合条件的索引对,则返回空列表。
if __name__ == "__main__":
# 输入读取
nums_input = input("Enter a list of integers separated by spaces: ")
target_input = int(input("Enter the target integer: "))
# 处理输入
nums = list(map(int, nums_input.split()))
# 解决方案和输出
sol = Solution()
result = sol.twoSum(nums, target_input)
print(f"The indices of the two numbers are: {result}")
num_map = {}
创建一个空的哈希表num_map
,用来存储每个数字对应的索引。for i, num in enumerate(nums):
使用enumerate
函数遍历列表nums
,同时获取每个元素的索引i
和元素值num
。complement = target - num
计算目标值target
与当前元素num
的差值,即另一个需要找到的数值。if complement in num_map:
检查差值complement
是否已经在num_map
中存在,如果存在,说明找到了符合条件的索引对,直接返回结果[num_map[complement], i]
。num_map[num] = i
将当前元素num
作为键,其索引i
作为值存入哈希表num_map
中,以便后续查找与之匹配的另一个数。- 如果遍历完整个列表
nums
都没有找到符合条件的索引对,则返回空列表[]
。
这种方法通过哈希表的快速查找,大大提高了解决问题的效率,时间复杂度为 O(n),空间复杂度也是 O(n)(因为需要额外的哈希表存储)