官方题目描述:
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。
解题思路:
这个题目主要的重点是拿到数值以及对应的下标,最后判断两数之和是否等于target
解法思路1 -- 使用哈希表(字典)+ enumerate函数
:
利用哈希表来存储已经遍历过的元素及其对应的索引。在遍历过程中,检查当前元素与目标值的差值是否已经在哈希表中,如果是则找到了答案,否则将当前元素及其索引添加到哈希表中。这种方法的时间复杂度为 O(n)。
使用python的内置函数,于同时获取列表、元组、字符串等可迭代对象的索引和对应的值。
如何使用 enumerate()
函数:
fruits = ['apple', 'banana', 'orange']
for index, fruit in enumerate(fruits, start=1):
print(f'Index {index}: {fruit}')
在这个示例中,enumerate(fruits, start=1)
返回一个枚举对象,对应关系如下:
- 第一次迭代:index=1, fruit='apple'
- 第二次迭代:index=2, fruit='banana'
- 第三次迭代:index=3, fruit='orange'
通过 enumerate()
函数,我们可以方便地获取可迭代对象的索引值,使代码更加简洁和易读。这在某些需要索引值的场景下非常有用,尤其是在处理列表、元组等数据结构时。
应用在这个题目上:
利用哈希表来存储已经遍历过的元素及其对应的索引,用enumerate函数将数值和索引值拿到,通过target与遍历的数值做减法,然后拿着差值去列表中查看这个值的索引,然后输出这两个值的索引。这个算法的时间复杂度为
O(n)。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
index_map = {}
for i, num in enumerate(nums):
complement = target - num
if complement in index_map:
return [index_map[complement], i]
index_map[num] = i
return []
解法思路2--暴力法:
最直接的方法是使用两层循环来遍历数组中的每对元素,然后检查它们的和是否等于目标值。这种方法的时间复杂度为 O(n^2)。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
n = len(nums)
for i in range(n):
for j in range(i+1, n):
if nums[i] + nums[j] == target:
return [i, j]
return []