LeetCode-Top100:两数之和(python)

发布时间 2023-04-16 21:39:31作者: 华东博客

 

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

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

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

 

示例 1:

输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 

示例 2:

输入:nums = [3,2,4], target = 6
输出:[1,2]

示例 3:

输入:nums = [3,3], target = 6
输出:[0,1]

  

提示:

  • 2 <= nums.length <= 104
  • -109 <= nums[i] <= 109
  • -109 <= target <= 109
  • 只会存在一个有效答案

 

以下是Python的程序实现:

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

解释:

此问题可以通过使用哈希表进行求解。我们可以创建一个字典,其中键是数组元素的值,值是数组元素的索引。我们遍历整个数组,并检查每个元素所对应的目标元素(target - num)是否已经在字典中。如果目标元素存在于字典中,则返回其索引和当前元素的索引。

时间复杂度:O(n)。我们只需遍历一次列表中的n个元素。在表中进行的每个查找都仅需O(1)的时间。

空间复杂度:O(n)。由于哈希表需要存储n个元素,因此需要额外的空间。