给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。 你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。 示例: 给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
解答:此题解法挺多的,一开始为了巩固学习enumerate()函数,专门用到enumerate()来解题,代码如下:
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
ls = []
for i, item in enumerate(nums):
#print(i, item)
if (target - item) in nums:
if (target - item) == item and nums.count(item) == 1:
continue
else:
ls.append(i)
return ls
执行用时 : 1720 ms, 在Two Sum的Python3提交中击败了34.96% 的用户
内存消耗 : 13.5 MB, 在Two Sum的Python3提交中击败了96.64% 的用户
这个执行时间吓到我了,所以必须优化,如下
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
d={}
size=0
while size < len(nums):
if target-nums[size] in d:
if d[target-nums[size]] < size:
return [d[target-nums[size]],size]
else:
d[nums[size]] = size
size = size + 1
执行用时 : 88 ms, 在Two Sum的Python3提交中击败了53.37% 的用户
内存消耗 : 14.2 MB, 在Two Sum的Python3提交中击败了40.75% 的用户
从执行时间上,已大大改进,但占用的内存也相比第一个较多。
原因是,第二种方法使用到的是字典,查找速度快,但占用内存大,第一种方法使用列表,查找速度慢,但占用内存小。
这也是很好的阐述python中字典dict和列表list的区别。