题目:https://leetcode.cn/problems/two-sum/
解题思路
哈希:访问值的复杂度为O(1)
python用字典来实现哈希操作,因为Python的字典采用了哈希表实现,最快能在O(1)时间内完成搜索
把数组的值放入字典的key中,数的序号放入value中,通过查找key,就可以快速找到该数的位置
注意:
字典的键值是唯一的,但是本题有个前提条件:只会存在一个有效答案
这意味着:如果两个相同的数a,a,那么target一定为a+a,因为如果target是a+b,那么就会有两个有效答案(有两个相同的a)
例如
输入:nums = [3,3], target = 6
输出:[0,1]
所以:
在放入一个数之前,判断target-num在不在数组中,有的话直接返回答案,不用放入字典
这样不会有两个相同的键值放入字典,且只需要遍历一次
代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap= {}
for i,num in enumerate(nums):
if target-num in hashmap:
return [hashmap[target-num],i]
hashmap[num]=i