题目
两数之和:给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
提示:你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
思路
1.首先想到的肯定是两个for循环遍历(超时淘汰)。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)):
for j in range(i+1,len(nums)):
if nums[i]+nums[j]==target:
return [i,j]
2.利用差值。因为a+b=target和different=target-a,所以different=b,且b可以在数组中找到。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)):
different=target-nums[i]
if(different in nums[:i]):
return [nums.index(different),i]
3.简化目前所用到的内置函数,index和切片,利用map,键对应nums[i],值对应i;每次遍历得出different再查找map的键里有没有和different所对应的键,不论有没有都将当前遍历的nums[i]存储到map里。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
nums_hash={}
for i in range(len(nums)):
different=target-nums[i]
if different in nums_hash:
return [nums_hash[different],i]
nums_hash[nums[i]]=i