LeetCode 算法题解:两数之和问题
本篇内容将探讨 LeetCode 平台上的经典算法问题——两数之和。我们将分析问题描述,并提供清晰易懂的代码解决方案,帮助您更好地理解和掌握该算法。
问题描述:
给定一个整数数组 nums
和一个目标值 target
,请你在数组中找出和为目标值的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
输入:nums = [2, 7, 11, 15], target = 9
输出:[0, 1]
解释:因为 nums[0] + nums[1] == 9 ,所以返回 [0, 1] 。
代码实现 (Python):
def two_sum(nums, target):
"""
使用哈希表解决两数之和问题。
Args:
nums: 整数数组。
target: 目标值。
Returns:
包含两个整数的列表,表示数组中和为目标值的两个数的下标。
"""
hash_map = {}
for i, num in enumerate(nums):
complement = target - num
if complement in hash_map:
return [hash_map[complement], i]
hash_map[num] = i
return []
代码解析:
- 创建一个空字典
hash_map
用于存储数组元素及其对应的索引。 - 遍历数组
nums
,对于每个元素num
,计算其需要的另一个数complement
。 - 检查
complement
是否已经在hash_map
中:- 如果存在,则返回
complement
的索引和当前元素num
的索引。 - 如果不存在,则将
num
及其索引添加到hash_map
中。
- 如果存在,则返回
- 如果遍历完数组还没有找到符合条件的两个数,则返回空列表。
时间复杂度: O(n),其中 n 是数组的长度。
空间复杂度: O(n),用于存储哈希表。