1.两数之和 Two Sum

最新推荐文章于 2024-09-13 18:11:35 发布

贪钱算法还我头发

最新推荐文章于 2024-09-13 18:11:35 发布

阅读量209

点赞数

文章标签： leetcode python

本文链接：https://blog.csdn.net/keiven_/article/details/111354933

版权

题目描述

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

示例：

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

来源：力扣（LeetCode）链接

第一次题解

第一次题解用了两层循环遍历，时间复杂度为 $O(n^2)$ 。执行用时：400 ms, 在所有 Python3 提交中击败了36.74%的用户；内存消耗：14.8 MB, 在所有 Python3 提交中击败了18.87%的用户。还是挺慢的。

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

最优题解

看了别人的答案，最优时间复杂度是 $O (n)$ ，用的字典。执行用时：32 ms, 在所有 Python3 提交中击败了98.10%的用户；内存消耗：14.9 MB, 在所有 Python3 提交中击败了18.00%的用户。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> 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