TwoSum python求解

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 参数,例如:


TwoSum python求解_第1张图片
例子.png

嗯嗯,看来 enumerate 还是很方便的嘛!学些了!

你可能感兴趣的:(TwoSum python求解)