leetcode(2)_1_easy_两数之和_python

最新推荐文章于 2021-11-21 16:20:22 发布

m_a_rookie

最新推荐文章于 2021-11-21 16:20:22 发布

阅读量70

点赞数

分类专栏： python_leetcode 文章标签： leetcode 数据结构 python 数组字典

本文链接：https://blog.csdn.net/qq_40137656/article/details/117987177

版权

python_leetcode 专栏收录该内容

109 篇文章 0 订阅

订阅专栏

两数之和

题目描述：
给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target 的那两个整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例：
输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
提示：

2 <= nums.length <= 10⁴
-10⁹ <= nums[i] <= 10⁹
-10⁹ <= target <= 10⁹
只会存在一个有效答案

解法

较为直接的思路是设置二层循环，但复杂度偏高。
由于题设中提到只有一个有效答案，所以可以在遍历的过程中，用一个字典记录下来各元素和target的差值；如果后续遍历时，某元素恰好在字典中，就说明该元素和前面的某元素即为问题解。因为要返回索引，所以字典的key-value是差值-索引。

代码

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        part = {}
        for index, item in enumerate(nums):
            if item in part:
                return [part[item], index]
            part[target - item] = index
        return []