力扣刷题——1.两数之和

最新推荐文章于 2024-11-15 16:27:37 发布

QHLJWM

最新推荐文章于 2024-11-15 16:27:37 发布

阅读量256

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/QHLJWM/article/details/126855577

版权

本文对比了寻找数组中两个数之和等于目标值的不同解决方案，包括暴力枚举、哈希表优化以及改进的两数之和思路。重点解析了Version1的改进和Version3的高效实现，展示了如何提升时间复杂度至O(N)并保持空间复杂度O(1)。

摘要由CSDN通过智能技术生成

Version 1

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        for i in range(len(nums)):
            remain = target - nums[i]
            if remain in nums:
                if nums.index(remain) != i:
                    return [i, nums.index(remain)]
        return []

自己最开始的想法，前两个实例都能运行出来，第三个出现了问题。

时间复杂度O(N^2)，空间复杂度O(1)。

Version 2 暴力枚举

官方给出的答案，解决了V1的问题。

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

优点：每个元素和前面的元素已经加和计算过了，只需要和后面的进行加和计算即可，这样能避免一个元素使用两次。

Version 3 哈希表

官方给出的答案。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        hashtable = dict()
        for i, num in enumerate(nums):
            if target - num in hashtable:
                return [hashtable[target - num], i]
            hashtable[nums[i]] = i
        return []

Version 4

自己改进V1

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        n = len(nums)
        for i in range(n):
            remain = target - nums[i]
            if remain in nums[i+1:n]:
                return [i, nums.index(remain)]
        return []

错误答案——修改未知