两数之和（Python，LeetCode）

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

TCatTime

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

阅读量496

点赞数

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

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

版权

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

摘要由CSDN通过智能技术生成

题目描述

给定一个整数数组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
        ret