python两数之和怎么求

983
2024/5/3 14:36:03
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

要求两个数的和,可以使用以下代码实现:

def two_sum(nums, target):
    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 None

nums = [2, 7, 11, 15]
target = 9
result = two_sum(nums, target)
if result:
    print(f"两数的索引为: {result}")
else:
    print("没有找到符合条件的两个数")

在这段代码中,two_sum 函数接受一个包含整数的列表 nums 和一个目标值 target 作为参数。函数通过遍历 nums 列表并将元素的值和索引存储在哈希表 hash_map 中,然后计算每个元素与目标值的差值 complement,检查是否存在该差值在 hash_map 中,如果存在则返回两个数的索引。如果遍历结束后没有找到符合条件的两个数,则返回 None

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: python中怎么重复循环程序