leetcode 第1题(两数之和)

153 阅读1分钟
给定一个整形数组,如果在数组中存在两数之和等于给定的目标值,则返回这两个数在数组中的索引 

你可以假定给定的输入里有且仅有一组解,同一元素不能被使用两次 

 例:

 输入:nums:[2, 7, 11, 15], target=9

 返回:[0, 1] 


 解:

class Solution:
    def twoSum(self, nums, target):
        if nums is None:
            return []
        d = {}
        for i in range(len(nums)):
            if target - nums[i] not in d:
                d[nums[i]] = i
            else:
                return d[target - nums[i]], i