给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
方法一:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
length = len(nums)
for i in range(length):
for j in range(i+1,length):
if(target == nums[i] + nums[j]):
return [i,j]
return []
用两层循环寻找解,最直观,但是时间复杂度高,为:O(n^2)
运行时间:3220ms,内存消耗:13.4mb
方法二:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
length = len(nums)
for i in range(length):
diff = target - nums[i]
if diff in nums:
if((nums.count(diff) == 1) & (i == nums.index(diff))):
continue
else:
return [i,nums.index(diff,i+1)]
return []
该方法的关键是直接判断差值是否在list里存在,只需要一个for循环。
运行时间:764ms,内存消耗:13.4mb
方法三:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
dic = {}
for index, value in enumerate(nums):
if ((target - value) in dic):
return [dic[target - value], index]
dic[value] = index
通过字典模拟哈希表。时间复杂度为O(n)。
运行时间:32ms,内存消耗:14.2mb
总结
该题的方法三体现了哈希的方法,可以通过空间换取更快的速度,此例中通过构造哈希表将遍历的时间复杂度从O(n^2)缩减至O(1)。
list.index(val) 获取元素对应在列表里的下标
list.index(val,index) 从下标index后获取val在列表里的下标
list.count(val) 获取该value在列表中的存在数量