领扣:两数之和

2020-06-17  本文已影响0人  领带衬有黄金

来源

https://leetcode-cn.com/problems/two-sum/

题目说明:

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

核心代码

def twoSum(nums: list, target: int):
    ret = {}
    for i, j in enumerate(nums):
        # 获取另一个值
        other_num = target - j
        if other_num in ret:
            return [ret[other_num], i]
        # 存储已经使用过的值
        ret[j] = i
    return None
上一篇 下一篇

猜你喜欢

热点阅读