每日一题:找出数组中的两个数,使其和等于给定的目标值。

解题思路:

  1. 创建一个空字典,用于存储数组中每个元素的索引。
  2. 遍历数组,对于每个元素:
  • 计算目标值与当前元素的差值 diff
  • 检查字典中是否存在 diff,如果存在,则找到了满足条件的两个数。
  • 如果不存在,则将当前元素及其索引添加到字典中。
  1. 如果没有找到满足条件的两个数,则返回一个空列表。
  2. 如果找到了满足条件的两个数,则返回它们的索引。
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 是数组的长度。在算法的执行过程中,我们只需要遍历一次数组,对于每个元素,可以在常数时间内进行查找操作。因此,该算法的时间复杂度是线性的。

猜你喜欢

转载自blog.csdn.net/qq_29669259/article/details/131591974