leetcode 力扣 001 两数之和 python

原始经典解法不再赘述，使用哈希表可以将时间复杂度降低到 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)（因为需要额外的哈希表存储）