1 两数之和
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
2 解题思路
核心:将目标值与列表中的元素做差,并将差值及其对应用来做做差的列表中的值的下标按字典保存,这样,下一次做差,如果差值已经在字典中,说明该列表元素值与已有差值之和能够等于目标值。取出差值为key值对应的value值及当前列表下标即可。
1、用目标值与依次列表中的每个值做差,做差后的结果用字典保存(key值为做差结果,value值为对应的列表下标),这样会获得了一个字典{‘a0’:0,‘a1’:1…‘a(i-1)’:(i-1),‘a(i)’:i},但这样做有个问题,如果目标值是当前列表值倍数,那么做差后,不管如何,差值都能在字典里找到,就会出现错误,因此,需要先对列表中的值在不在字典中做一个判断,这样每次字典中保存的数据为{‘a0’:0,‘a1’:1…‘a(i-1)’:(i-1)}
2、如果列表中第i个数的值能够在之前做差保存的字典中找到,那么就返回列表中第i个值在字典中对应的value值,及当前下标。
方法一:
class Solution:
def twoSum(self,nums: List[int], target: int) -> List[int]:
d = {}
for i in range(len(nums)):
a = target - nums[i]
if nums[i] in d:
return d[nums[i]],i
else:
d[a] = i
**注:LeetCode不需要调用函数,直接运行代码就可**以。
方法二:
class Solution:
def twoSum(self,nums: List[int], target: int) -> List[int]:
d = {}
for i,m in enumerate(nums):
a = target - m
if m in d:
return d[m],i
else:
d[a] = i
用枚举的方法,在同时需要取出下标和元素的情况下,速度会更快一点。