给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
这道题目不难,就是在给定的数组中找到两个数,两数之和要等于target。
完整代码如下所示:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
records = dict()
for index, value in enumerate(nums):
if target - value in records: # 遍历当前元素,并在map中寻找是否有匹配的key
return [records[target- value], index]
records[value] = index # 如果没找到匹配对,就把访问过的元素和下标加入到map中
return []
首先,先定义一个空字典records。接着,利用enumerate函数,对数组中的数进行枚举标号,以nums[2,7,11,15]为例,在经过enumerate函数后,index为0则其对应的value为2,index为1则其对应的value为7,index为2则其对应的value为11,index为3则其对应的value为15。
所以先判断target-value在不在字典里面,在的话则代表找到了和为target的两数,return [records[target- value], index],为什么records[target- value]在前呢,是因为该数已经在字典里了,代表它是之前就被遍历过了。
其实,代码并不难,甚至属于简单,但在学习过程中,我一开始一直有一个疑惑,records[value] = index这行代码为什么不是records[index] = value?一般不都是通过键来查询值吗?
后来我发现,之所以是用records[value] = index是为了提高查找效率。如果利用records[index] = value,字典中存储的是索引和值的对应关系。要查找补数是否存在,就需要遍历字典中的所有键值对,时间复杂度从O(1)变为了O(n)。