题目描述
https://leetcode-cn.com/problems/two-sum/
CornerCase
数组中出现重复的数字,比如nums=[2,2,3],target=4
同一元素不可重复使用
解决方案
1、暴力法
class Solution(object):
def twoSum(self, nums, target):
for i,num in enumerate(nums):
if target-num in nums[i+1:]:
return [i, nums[i+1:].index(target-num) + i + 1]
时间复杂度O(N2),空间复杂度O(1)
执行用时700ms,战胜50.62%的python代码提交
消耗内存13.4M,战胜95%的python代码提交
此处,index(x)方法返回列表中第一个出现的x的索引,因此要从num后开始找符合要求的索引
2、哈希(先建表,再查找)
针对暴力法时间复杂度高的问题进行优化,使用哈希,以空间换时间可以降低时间复杂度
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
dic = {}
for i,num in enumerate(nums):
dic[num] = i
for i,num in enumerate(nums):
value = dic.get(target-num)
if value is not None and value!=i:
return [i, value]
时间复杂度O(N),空间复杂度O(N)
执行用时64ms,击败了82.5%的python代码
内存消耗15.6M,击败了9.66%的python代码
此处,由于建立哈希表时只保留相同键的最后一个键值对,因此当出现cornercase时,哈希表中的num跟列表中的num若之和为target,则此两个num索引必不同
3、哈希(边查找边建表)
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
dic = {}
for i,num in enumerate(nums):
if dic.get(target-num) is not None:
return [dic.get(target-num),i]
dic[num] = i