题目
给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例 :
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
自己的想法
算法实现
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
j = len(nums)
for i in range(j):
t = target - nums[i]
if t in nums[i + 1:]:
return [i, nums.index(t, i + 1)]
else:
continue
执行结果
执行结果 : 通过
执行用时 : 868 ms, 在所有 Python3 提交中击败了36.49%的用户
内存消耗 : 14.1 MB, 在所有 Python3 提交中击败了68.74%的用户
复杂度分析
-
时间复杂度:O(n^2),
我们只遍历了包含有 n 个元素的列表一次。在表中进行的每次查找花费O(n)的时间。 -
空间复杂度:O(n),
所需的额外空间取决于range中存储的元素数量,最多需要存储 n 个元素。
哈希表法
class Solution:
hashmap = {}
for index, num in enumerate(nums):
if target - num in hashmap:
return [index, hashmap[target - num]]
hashmap[num] = index
执行结果
复杂度分析
-
时间复杂度:O(n),
我们只遍历了包含有 n 个元素的列表一次。在表中进行的每次查找只花费 O(1) 的时间。 -
空间复杂度:O(n),
所需的额外空间取决于哈希表中存储的元素数量,该表最多需要存储 n 个元素。
小结
先按照自己的想法设计,通过判断target - nums[i]
是否存在于后面的列表中来找到另一个数字,算法并不是很快。之后学习了哈希表法,通过建立字典,将键与num1匹配,值与num1的索引匹配,一边遍历数组,一边查找并将键值加入字典中,由于哈希表查找时间复杂度低,所以会快很多。