欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

leetcode1 两数之和 python

程序员文章站 2022-07-14 17:29:02
...

题目:

给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。

思路:

利用哈希快速查找。
实现就是用一个字典,键是数字大小,值时序号。
遍历nums,如果target-nums[i]在字典里,读出对应序号就可以和i一起返回了。

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        va2index = {}
        for i in range(len(nums)):
            if target-nums[i] in va2index:
                return [va2index[target-nums[i]],i]
            va2index[nums[i]] = i