给定一个整形数组,如果在数组中存在两数之和等于给定的目标值,则返回这两个数在数组中的索引
你可以假定给定的输入里有且仅有一组解,同一元素不能被使用两次
例:
输入: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