版权声明:本人原创,转载请注明出处 https://blog.csdn.net/zjn295771349/article/details/86572602
给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
我的答案:
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for i, num1 in enumerate(nums):
for j, num2 in enumerate(nums[i+1:]):
if num1 + num2 == target:
return i,i+j+1
我用的方法是暴力搜索,遍历nums中的每对儿数,计算每对儿之和,判断是否等于target,结果是正确的但是效率低。
时间复杂度:O(n2)、空间复杂度:O(1)
下面的方法是网上找的比较好的:
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
n = len(nums)
#创建一个空字典
d = {}
for x in range(n):
a = target - nums[x]
#字典d中存在nums[x]时
if nums[x] in d:
return d[nums[x]],x
#否则往字典增加键/值对
else:
d[a] = x
可以看出来用时极短,这是因为这个方法是先计算出target与nums中第一个数的差值,把差值存进字典里作为键、索引作为值,第一次循环理解:d[7]=0 即字典d={7:0},表示为索引0需要数组里值为7的元素配对。 if 判断是否为前面元素所需要配对的值 , 是则返回两个索引值。
时间复杂度:O(1) 、空间复杂度O(n) (补充:dict对象的存储结构采用的是散列表(hash表),其在最优情况下查询复杂度为O(1))。
参考资料:https://www.cnblogs.com/bibi-feiniaoyuan/p/9260172.html