题目链接:https://leetcode-cn.com/problems/two-sum/
解法一:遍历数组,找当前元素与当前元素后面的元素之和是否满足条件。
时间复杂度O(N^2)
def twoSum(self, nums: List[int], target: int) -> List[int]:
nums_list = enumerate(nums)
for idx,n in nums_list:
if (target - n) in nums[idx+1:]:
return [idx,nums[idx+1:].index(target-n)+idx+1]
return None
1.ennunrate函数:
enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标。
seasons = ['Spring', 'Summer', 'Fall', 'Winter']
list(enumerate(seasons))
list(enumerate(seasons, start=1)) # 下标从 1 开始
输出结果:
[(0, 'Spring'), (1, 'Summer'), (2, 'Fall'), (3, 'Winter')]
[(1, 'Spring'), (2, 'Summer'), (3, 'Fall'), (4, 'Winter')]
解法二:借助哈希表 由于哈希查找的时间是O(1) 牺牲空间节约时间
https://leetcode-cn.com/problems/two-sum/solution/jie-suan-fa-1-liang-shu-zhi-he-by-guanpengchn/ 有动画演示
思路:遍历数组 nums,i 为当前下标,每个值都判断map中是否存在 target-nums[i] 的 key 值
如果存在则找到了两个值,一个是i,另一个是key值对应的value值,也就是在数组中key对应的下标,如果不存在则将当前的 (nums[i],i) 存入 map 中,继续遍历直到找到为止。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap = {}
for index,num in enumerate(nums):
hashmap[num] = index
for i,num in enumerate(nums):
j = hashmap.get(target-nums[i])
if j is not None and i!=j:
return [i,j]