题目:两数之和
right answer
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
hashmap = dict()
for idx,n in enumerate(nums): # 其实遍历一次就够了
if target-n in hashmap: # 两数之和,两个数可以一样,但角标不可一样
return [hashmap[target-n], idx] if hashmap[target-n] < idx else [idx,hashmap[target-n]] # 确保更小的角标在前
hashmap[n] = idx
return [] # 无答案
思路
两数之和,要想到这是个“查找”问题,那么使用哈希表就比纯数组要好很多,毕竟数组的查找是遍历,复杂度O(N),而哈希表是O(1)!
此外,直觉上使用哈希表是要循环2次,一次是dict的存储(元素需要一个个地被存入),一次是遍历dict进行查找。但由于哈希表查找速度是O(1),因而其实还是O(N)的复杂度;
但实际上不需要两次循环。因为有答案的结果一定是成对出现的,所以只要追溯到了后者,前者就一定会被“捞出来”。
最后,12行的赋值代码必须不能在if判断之前,以确保nums数组里相同的数值不会被dict赋值自动抹除