题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
链接:https://leetcode-cn.com/problems/two-sum
1.示例
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
2. 解析
最简单的方式就是使用暴力破解,遍历数组的每一个元素x,寻找数组中存在另一个值等于target-x。
#伪代码如下
for i in range(len(L)):
for j in range(len(L)):
if(L[i]+L[i] == target):
pass
从代码上能够看出,这段代码的时间复杂度为:O(n2)。因此这种方式就存在一个问题,当数组过长时,会出现超时的情况。
因此我们选择使用哈希表的方式对数组进行查找。
哈希表也叫做散列表,他是通过某种算法将要检索的项与用来查找的索引建立联系,以加快查找的速度。简单来说哈希表是根据<key,value>的形式,进行直接访问的数据结构。
Python中的字典就是利用哈希表实现的,因此我们使用字典模拟哈希存储过程,提高查询速度。
先将数组存为字典形式,然后从字典中查找元素x,和target-x。代码实现如下所示:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap = {} # 字典模拟哈希
for key,value in enumerate(nums):
hashmap[value] = key
for i,j in enumerate(nums):
cha = hashmap.get(target - j)
if cha is not None and i != cha:
return i,cha
3. 测试用例
用例编号 | 测试输入 | 实际输出 |
---|---|---|
test_001 | List = [1,2,3,4,5]; target = 3 | (0,1) |
test_002 | List = [-1,-2,-3,-4];target = -3 | (0,1) |
test_003 | List=[-1,0,3,-4,5];target = 2 | (0,2) |
test_004 | List=[1,2,3,4,5];target = 0 | None |
test_005 | List=[-1,-2,-3,-4,-5];target = 0 | None |