LeetCode Python3——1. 两数之和

问题描述

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

Sol 1:

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        index=[]
        for f_index,i in enumerate(nums):
            for  s_index,j in enumerate(nums[f_index+1:]):
                if i + j ==target:
                    index.append(f_index)
                    index.append(f_index + s_index + 1)
                    return index

方法1是很直接的想法，用两次循环将每两个数依次相加，$enumerate()$函数可以获得$nums$对应的索引，时间复杂度为$O(n^2)$，空间复杂度为$O(1)$，运行时间为6692ms，内存占用为7.2M。

Sol 2:

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

方法2是比较巧妙的想法了，只用了一次循环，在进行迭代并将元素插入到表中的同时，我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在，那我们已经找到了对应解，并立即将其返回。
时间复杂度：O(n)， 我们只遍历了包含有 n 个元素的列表一次。
空间复杂度：O(n)， 所需的额外空间取决于哈希表中存储的元素数量，该表最多需要存储 n 个元素。
运行时间为44ms，内存占用为7.9M。

结论

方法1：时间复杂度高，空间复杂度稍低
方法2：空间复杂度稍高，时间复杂度很低