两数之和（Python，LeetCode）

最新推荐文章于 2023-10-22 23:28:26 发布

TCatTime

最新推荐文章于 2023-10-22 23:28:26 发布

阅读量528

点赞数

CC 4.0 BY-SA版权

分类专栏： Python # Python常见题目文章标签： python leetcode

本文链接：https://blog.csdn.net/TCatTime/article/details/106646920

Python 同时被 2 个专栏收录

128 篇文章

订阅专栏

Python常见题目

46 篇文章

订阅专栏

这篇博客讨论了如何使用Python解决LeetCode上的两数之和问题。通过建立字典来存储数组元素和其索引，在一次遍历中找到目标和的两个数的索引。代码简洁高效，时间复杂度为O(n)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个整数数组nums和一个目标值target，请你在该数据中找出和为目标值的那两个整数，并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

输入/输出示例：

输入条件	给定nums = [2, 7, 11, 15], target = 9
返回值	[0, 1]
解释	因为nums[0] + nums[1] = 2 + 7 = 9，所以返回[0, 1]

解决方案

题目的要求是返回线性表（List）中和为target的两个数字的下标。在这个题目中，线性表中每个元素有两个需要关注的属性：元素本身的数值和元素的下标。由于题目中明确同一个元素不会使用两遍，因此可以明确元素在线性表中是唯一的。我们可以使用字典（dict）和一次线性表遍历来找到答案。字典中存储线性表中的元素值与下标的映射，即以元素值为key，下标为value。在每次线性表遍历中，若target与遍历元素差的结果在字典中存在，则说明找到了最终结果，否则将本次遍历的元素值与元素下标作为键值对存入字典中。

代码

class Solution:
    def twoSum(self, nums, target):
        nums_dict = {}
        for i in range(len(nums)):
            if target - nums[i] in nums_dict:
                return [nums_dict[target - nums[i]], i]
            nums_dict[nums[i]] = i
        return []


# example
if __name__ == '__main__':
    solution = Solution()
    print(solution.twoSum([2, 7, 11, 15], 9))

代码分析

class Solution:
    def twoSum(self, nums, target):
        # 声明一个dict，用来存储线性表中元素值与元素下标映射关系
        nums_dict = {}

        # 开始遍历线性表
        for i in range(len(nums)):
            # 如果target - nums[i]在dict中存在，则说明找到了两个和为target的元素
            if target - nums[i] in nums_dict:
                return [nums_dict[target - nums[i]], i]

            # 否则将本次遍历的元素信息（元素值和下标）存入dict中，供后续的遍历查找
            nums_dict[nums[i]] = i

        # 没有找到和为target的两个元素，返回一个空list
        return []


# example
if __name__ == '__main__':
    solution = Solution()
    print(solution.twoSum([2, 7, 11, 15], 9))