1.问题说明
来源,给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
2.原理说明
这个问题其实比较简单,目标是找到两个索引i
和j
,使得num[i] + num[j] = target
,那么这个问题中其实三个数中我们知道其中两个,另外的一个就是已知的了,好按照这个思路我们可以按照下面的步骤:
- 1.定义一个字典,格式:
{index:value}
,目的是用来存储值 - 2.遍历所有的元素,并且用
target - num[i] = num[j]
,其实此时前面两个都是知道的,但是我们不知道后面的num[j]
在数组中有没有 - 3.判断
num[j]
有没有在字典中,如果有那么就是找到了,直接返回即可 - 4.如果没有在字典中,那么就放入字典中,按照格式放入
3.代码说明
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
# 定义一个字典
hashmap = {}
# 将数组封装成一个枚举类型,并遍历
for index , value in enumerate(nums):
# 如果目标值和当前值得差在字典中,说明找到了两数之和,返回两数的索引即可
if target - value in hashmap:
return [index, hashmap[target - value]]
# 如果当前值和目标值的差没有在字典中,那么就以当前值为为key,索引为value
hashmap[value] = index