给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
第一次,内容无误,但是超时了
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)):
for j in range(len(nums)):
if i != j:
if nums[i] + nums[j] == target:
return i,j
第二次,同样超时
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)-1):
for j in range(i+1,len(nums)):
if nums[i] + nums[j] == target:
return i,j
第三次,仿照官方的java题解,改写,成功通过(暴力法),执行时间5876 ms
However,当我第二次使用这种做法是,结果又变成了超出时间限制
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)-1):
for j in range(i+1,len(nums)):
if nums[j] == target - nums[i]:
return i,j
第四次,仿照题解中的一个C++视频讲解,实现其思路,执行时间60ms
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
# 定义一个字典
result = {}
for i in range(len(nums)):
temp = target - nums[i]
if temp not in result:
result[nums[i]] = i
else:
return result[temp],i
第五次,观看了另一个大神的解法,对第四次进行简单改善(几乎不变) ,执行时间84ms,主要是为了记录enumerate的用法
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
result = {}
for i,j in enumerate(nums):
temp = target - j
if temp in result:
return [result[temp],i]
result[j] = i
(enumerate用法:https://docs.python.org/zh-cn/3/library/functions.html#enumerate)
个人理解:enumrate用法就是把每个位置和其对应的元素以元组的形式表现出来
这道题还有好多种解法,后续会持续更新。