leetcode 算法题,传送门
例子:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
# 注意:只有唯一解
乍一看,没有经验的一定会用到两重循环,比如这样:
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for i in range(len(nums)):
for j in range(i+1, len(nums)):
if nums[i] + nums[j] == target:
return [i,j]
但是如果有经验的人就会思考了,上面的程序时间复杂度是 O(n^2),于是在深思熟虑之后写下了只用一重循环的代码,用到了字典这一基本数据结构,仔细想想,如果只用判断 target - x == y
即可判断。仔细想想可以自己写出来。
但是我偶然在 stackoverflow 上遇到了一种更好的写法:
class Solution(object):
def twoSum(self, nums, target):
keys = {}
for cnt, num in enumerate(nums):
if target - num in keys:
return keys[target-num], cnt
keys[num] = cnt
这里面用到了 enumerate ,我查了一下,这个内置函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标。
它还提供了一个 start 参数,例如:
例子.png
嗯嗯,看来 enumerate 还是很方便的嘛!学些了!
网友评论