https://leetcode.cn/problems/two-sum/
题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出和为目标值 target 的那两个整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
解题思路及代码
解法一:两重for循环
思路:先拿出列表里的第0个数字,依次尝试和第1个、第2个……第n个相加,看能否等于目标数字,如果有某个组合等于目标数字,就返回这个组合的两个索引值,如果都不行,再拿出第1个数字,依次尝试和第2个、第3个……第n个相加,看能否等于目标数字……每个数字都不需要回过头和前面的数字相加,因为之前已经尝试过了
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)):
for j in range(i+1, len(nums)):
if nums[j] == target - nums[i]:
return [i, j]
else:
return "No Exists"
两重循环,时间复杂度就是O(n^2),其中n=len(nums)
解法二:单重for循环+字典存储值(key)和索引(value)
思路:先建立一个用于存放数字(key)和索引(value)的空字典。从列表中取出一个数字,然后看字典里是否存在能跟这个数字相加得到目标数字的数字。如果存在,就返回两个数字的索引,不存在,就把取出来的这个数字作为键,它的索引作为值存进字典,并取出下一个数字,重复刚才的操作,直到找到合适的两个数字或者全部数字都看过一遍为止
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
dict = {} # 定义一个空字典
for i, value1 in enumerate(nums):
value2 = target - value1
if dict.get(value2) != None: # value2作为字典的key
return [dict[value2], i] # dict[value2]为该值对应的索引,即数组下标
dict[value1] = i # 若没找到,就把这个值和下标存进字典
else:
return "No Exists"
扫描一次就能得到结果,时间复杂度就是O(n),其中n=len(nums)
知识点总结
(一)enumerate()函数
用法:用来遍历一个集合对象,它在遍历的同时还可以得到当前元素的索引位置
names = ["Alice","Bob","Carl"]
for index,value in enumerate(names):
print(f'{index}: {value}')
运行结果:
0: Alice
1: Bob
2: Carl
(二)dict.get()
用法:get()函数用于获取字典中对应key的value。第一个参数即为key,第二个参数设置的是字典中不包含该键时,get()函数的返回值。若不设置第二个参数,则默认返回None
d = {'a':2, 'b': 3}
d.get('a')
d.get('c')
d.get('f', 5)
运行结果:
2
None
5