题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定nums=[2,7,11,15],target=9,因为nums[0]+nums[1]=9,所以返回[0,1]和与其相对应的数值2,7。
代码
from typing import List
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
if len(nums)<=1:
return False
else:
for i in range(0, len(nums)):
for j in range(i+1, len(nums)):
if target==nums[i]+nums[j]:
b=[]
b.append(i)
b.append(j)
return b
nums = [2,7,11,15]
target = 9
a = Solution()
c=a.twoSum(nums, target)
print(c)
print(nums[c[0]],nums[c[1]])
注意:这种算法使用的是嵌套循环遍历所有可能的数字对,算法复杂度为O(n2),根据提示显示这样虽然可以保证数据的完整性,但是运行时间过长,所以下面对算法进行了优化.
from typing import List
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
n=len(nums)
result={}
if n<=1:
return False
else:
for i in range(0,n):
if nums[i] in result:
return result[nums[i]],i
else:
result[target-nums[i]]=i
nums = [3,2,4]
target = 6
a = Solution()
c=a.twoSum(nums, target)
print(c)
print(nums[c[0]],nums[c[1]])
总结:
后者的算法,使用Python字典的key-value进行的匹配,算法复杂度为O(n).思路是将目标值与数组元素作差,将差值作为key写入字典中,利用一次循环,匹配数组中元素和字典中差值,返回数组下标及相对应的整数。