leetcode 1. 两数之和 python解法

遍历数组,判断目标和当前数的差是否为前面出现过的数(存在字典中),如果不存在则将当前数和下标存在字典中,存在则返回当前下标和字典中存的下标。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dic = {}
        for i in range(len(nums)):
            if target - nums[i] in dic:
                return [i, dic[target - nums[i]]]
            else:
                dic[nums[i]] = i

你可能感兴趣的:(leetcode,leetcode,python,算法)