给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
1. 暴力解法
从列表的第一项开始,取它后面的数,判断是否等于target-nums[i]
返回 i , j
class Solution():
def twoSum(self, nums, target):
for i in range(len(nums)-1):
sub = target-nums[i]
for j in range(i+1,len(nums)):
if nums[j]==sub:
return [i,j]
执行用时:3476 ms
内存消耗:14.6 MB
2. 利用一个for循环:
从列表的第一项开始,判断target-nums[i]
是否存在与列表中,不存在的话就继续下一次循环,存在的话就判断索引值是否与 i 相等,相等的话就进行下一次循环,不相等的话就返回[ i , j ]
class Solution():
def twoSum(self, nums, target):
for i in range(len(nums)):
sub = target-nums[i]
if sub not in nums:
continue
j=nums.index(sub)
if i==j:
continue
return [i,j]
执行用时:1064 ms
内存消耗:14.9 MB
- 利用字典
class Solution():
def twoSum(self, nums, target):
dic={}
for i,num in enumerate(nums):
dic[num]=i
for i in range(len(nums)):
sub = target-nums[i]
if sub not in dic.keys():
continue
j=dic[sub]
if i==j:
continue
return [i,j]
执行用时:72 ms
内存消耗:15.4 MB
- 对上个解法稍作改进
将if sub not in dic.keys()
改为if sub in dic.keys()
,将if i==j
改为if i!=j
可以少做几次判断,提高运行速度。
class Solution():
def twoSum(self, nums, target):
dic={}
for i,num in enumerate(nums):
dic[num]=i
for i in range(len(nums)):
sub = target-nums[i]
if sub in dic.keys():
j=dic[sub]
if i!=j:
return [i,j]
执行用时:64 ms
内存消耗:15.4 MB
- 字典解法的改进
参考答案中大佬们的解法,发现没有必要生成完整的字典,只需要生成nums[i]
前面的字典就可以了,可以少用一个for循环。
class Solution():
def twoSum(self, nums, target):
dic={}
for i in range(len(nums)):
sub=target-nums[i]
if sub in dic:
return i,dic[sub]
dic[nums[i]]=i
#字典开始为空,从列表的第一个元素开始,判断target - nums[x]
是否存在字典中。判断时,nums[i]
还未在字典中,所以不会产生一值多用的情况。如果sub
不在dic
中,就将i
和nums[i]
存放在字典中,继续选择列表的下一个值进行判断。相当于判断nums[i]
与其之前的值是否存在加和为target
的结果。
执行用时:64 ms
内存消耗:15.1 MB