每天练习python-两数之和

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

思路

我们可以用哈希表来优化暴力枚举的过程。具体地，在遍历数组 nums 时，对于每个元素 x，我们首先查询哈希表中是否存在 target - x。如果存在，则我们已经找到了两个元素的和为目标值，直接返回对应的数组下标即可；如果不存在，则将 x 插入到哈希表中，等待后续的查询。

代码实现

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        hashmap = {}
        for i, num in enumerate(nums):
            if target - num in hashmap:
                return [hashmap[target - num], i]
            hashmap[num] = i

在上面的代码中，我们使用一个字典（即哈希表） hashmap 来存储每个元素及其下标。具体地，键为元素的值，值为元素的下标。这样，在遍历数组 nums 时，我们只需要查询字典中是否已经存在 target - num 即可。如果存在，则返回对应的值和当前下标；如果不存在，则将当前元素及其下标插入哈希表中等待后续的查找。由于哈希表的查询操作是常数时间复杂度的，因此总时间复杂度为 O(n)，空间复杂度也为 O(n)。