LeetCode 算法题解:两数之和问题

boiler3964 5 0 java 2024-07-05 02:07:47

本篇内容将探讨 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 []

代码解析:

  1. 创建一个空字典 hash_map 用于存储数组元素及其对应的索引。
  2. 遍历数组 nums,对于每个元素 num,计算其需要的另一个数 complement
  3. 检查 complement 是否已经在 hash_map 中:
    • 如果存在,则返回 complement 的索引和当前元素 num 的索引。
    • 如果不存在,则将 num 及其索引添加到 hash_map 中。
  4. 如果遍历完数组还没有找到符合条件的两个数,则返回空列表。

时间复杂度: O(n),其中 n 是数组的长度。

空间复杂度: O(n),用于存储哈希表。

用户评论
请输入评论内容
评分:
暂无评论