LeetCode练习1——数组（两数之和）

最新推荐文章于 2022-03-29 11:41:52 发布

原创最新推荐文章于 2022-03-29 11:41:52 发布 · 121 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #python

LeetCode 专栏收录该内容

85 篇文章

订阅专栏

这篇文章介绍了如何使用Python解决LeetCode中的经典问题——在给定整数数组中找到两个数相加等于目标值的下标。通过字典实现，提供了示例和解析。

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

示例 1：
输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：
输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：
输入：nums = [3,3], target = 6
输出：[0,1]

提示：
2 <= nums.length <= 103
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案

解答：用字典

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        dict_num = {}
        for i, v in enumerate(nums):
            tmp = target - v
            if dict_num.get(tmp) is not None:
                return i, dict_num[tmp]
            else:
                dict_num[v] = i