两数之和

合集 - leetcode(5)

1.两数之和2023-04-06

2.两数相加2023-04-063.N 字形变换2023-04-154.最长回文子串2023-04-155.寻找两个正序数组的中位数2023-04-15

收起

题目介绍#

难度简单

给定一个整数数组 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]

题解#

暴力解时间复杂度为$O(n^2)$，使用哈希表优化时间复杂度为$O(n)$。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        num_dict = {}
        for idx, val in enumerate(nums):
            if (target - val) not in num_dict.keys():
                num_dict[val] = idx
            else:
                return [num_dict[target - val], idx]