一、O(n^2)
采用两层嵌套for循环,这里不做解释
二、O(n)
前提:1.python中哈希表类似与JavaScript中的对象
2. enumerate
是一个内置函数,用于在迭代过程中同时获取元素的索引和值。它接受一个可迭代对象作为参数,并返回一个生成器对象,每次迭代时都会返回一个包含索引和对应值的元组。
思路:将列表([ ])中数据转化成哈希表({ val : key })的形式,然后遍历一遍寻找与另一个相加为target的数。
问题: 1.如果先存再找会出现如下情况
nums = [0, 0] target = 0
ans = [0, 0]
原因是当前是0,寻找target,程序寻找hash表,结果找的正是当前0的下标,所以返回错误的结果,解决方法先找再存。
2.如果先边遍历边存哈希表会出现如下情况
nums = [3, 3] target = 6
ans = [1,1]
原因是哈希表会覆盖前面相同 val 的 key 值
综上所述我们应该先找后将列表存为hash表,题解如下
def twoSum(nums, target):
# 定义hash表
index_dict = {}
# 遍历nums
for i, num in enumerate(nums):
# 寻找相加为target的元素
mid = target - num
if mid in index_dict:
return [index_dict[mid], i]
index_dict[num] = i
return []