Python实现与LeetCode--数组

最新推荐文章于 2022-04-17 10:08:17 发布

转载最新推荐文章于 2022-04-17 10:08:17 发布 · 125 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/pankypan/p/11099657.html

文章标签：

#python

本文详细介绍了在LeetCode上解决“两数之和”问题的两种方法，第一种方法通过两次循环实现，而第二种方法则进行了一次循环优化，利用字典结构在O(n)的时间复杂度内找到目标和的两个数。文章提供了完整的Python代码示例，展示了如何使用字典以空间换时间，提高查找效率。

Python实现与LeetCode--数组

LeetCode

两数之和

方法一：两遍循环

以空间换取时间，循环这个nums，用字典（hash table）来记录nums,以nums的值为key，对应的值的数组下标为value（此时当数组内有相同的值时，字典会自动舍去多余的只保留一个）

第二次循环这个nums，求出 complement = target - nums[i]，判断complement是否作为key存在这个字典中，并判断这个complement是否是nums[i]，即本身

class Solution:
    def twoSum(self, nums, target):
        nums_dict = {}
        for index, v in enumerate(nums):
            nums_dict[v] = index  # 注意，此方式会自动消掉nums相同的元素，只保留一个

        for i in range(len(nums)):
            complement = target - nums[i]
            if complement in nums_dict and nums_dict.get(complement, None) != i:
                return [i, nums_dict.get(complement)]
        return None


if __name__ == "__main__":
    nums = [1, -8, 7, 4, 6]
    solution = Solution()
    res = solution.twoSum(nums, -4)
    print(res)

方法二：一遍循环

class Solution:
    def twoSum(self, nums, target):
        nums_dict = {}
        length = len(nums)
        for i in range(length):
            complement = target - nums[i]
            if complement in nums_dict:
                return [i, nums_dict.get(complement)]
            else:
                nums_dict[nums[i]] = i
        return None


if __name__ == "__main__":
    nums = [2, 7, 11, 15]
    solution = Solution()
    res = solution.twoSum(nums, 9)
    print(res)

复杂度分析：两种方法复杂度是一样的，方法二是方法一的一种优化