题目及解题思路来源:LeetCode
1. 两数之和
问题: 给定一个整数数组 n u m s nums nums和一个目标值 t a r g e t target target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9,所以返回 [0, 1]
解决方法
1. 暴力法
遍历每个元素 x x x,并查找是否存在一个值与 t a r g e t − x target-x target−x相等的目标元素
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: 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]
return None
复杂度分析:
- 时间复杂度: O ( n 2 ) O(n^{2}) O(n2),对于每个元素,我们试图通过遍历数组的其余部分来寻找它所对应的目标元素,这将耗费 O ( n ) O(n) O(n)的时间。因此时间复杂度为 O ( n 2 ) O(n^{2}) O(n2)。
- 空间复杂度: O ( 1 ) O(1) O(1)
2. 两遍哈希表
为了对时间复杂度进行优化,我们需要一种更有效的方法检查数组中是否存在目标元素,我们需要找出它的索引。保持数组中的每个元素与其索引相互对应的最好方法是什么?哈希表。
通过以空间换取速度的方式,我们可以将查找时间从 O ( n ) O(n) O(n)降到 O ( 1 ) O(1) O(1)。哈希表正是为此目的而构建的,它支持以近似恒定的时间进行快速查找。“近似”原因是一旦出现冲突,查找用时可能会退化到 O ( n ) O(n) O(n)。但是只要仔细地挑选哈希函数,在哈希表中进行查找的用时应当被摊销为 O ( 1 ) O(1) O(1)。
两次迭代。第一次迭代,将每个元素的值和它的索引添加到表中。然后,在第二次迭代中,检查每个元素对应的目标元素 ( t a r g e t − n u m s [ i ] ) (target-nums[i]) (target−nums[i])是否存在表中。注意,该元素不能是 n u m s [ i ] nums[i] nums[i]本身!
# 利用字典实现哈希表
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]