Python学习笔记——enumerate（）函数

enumerate（）函数的描述

enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列，同时列出数据和数据下标，一般用在 for 循环当中。

使用语法：

enumerate(sequence,[start=0])

参数

• sequence 一个序列、迭代器或其他支持迭代的对象 （string、list 和 tuple 都属于 sequence（序列））
• start 下标起始位置。默认为0
  返回值
  返回enumerate对象

案例1：

l1 = [11,22,33,44,55]
em = enumerate(l1)
l2 = [i for i in em]
print(l2)

输出结果：
[(0, 11), (1, 22), (2, 33), (3, 44), (4, 55)]

案例2：

l1 = [11,22,33,44,55]
em = enumerate(l1,start = 100)
l2 = [ i for i in em]
print(l2)

输出结果：
[(100, 11), (101, 22), (102, 33), (103, 44), (104, 55)]

实战应用

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:
给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

普通解法

大多数人会想到下面这种解法，但时间复杂度为O(n^2)

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        rtype = []
        for i in range(len(nums)-1):
            for j in range(i+1,len(nums)):
                if nums[i]+nums[j]==target:
                    return [i,j]

使用enumerate（）函数

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        hash_map =dict()
        for i,x in enumerate(nums):
            if target - x in hash_map:
                return [i,hash_map[target-x]]
            hash_map[x] = i

使用此种解法时间复杂度得到了一定的优化，复杂度为O(n)
大家可以通过这几个小例子去深刻体会一下enumerate()函数的用法

最后的实战应用题是我在LeetCode网碰到的，正是因为这道题，我才记录下这篇博客。
注：LeetCode是一个美国的在线编程网站，上面主要收集了各大IT公司的笔试面试题，对于找工作是一个不可多得的好帮手。