【LeetCode HOT100】1. 两数之和

加载中... 浏览

题目

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

提示

一句话思路:用哈希表记录「每个数字出现过的下标」,遍历时查找 target - 当前数 是否已经在哈希表中。

  • 为什么用哈希表:能把「查找另一个数」从 O(n) 降到 O(1),整体时间复杂度 O(n)。
  • 为什么边遍历边存:保证不会使用两次相同元素(先查再存,避免把当前元素自己当成答案)。

答案

python
class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dct = {}                    # 哈希表:记录「值 -> 下标」,用于 O(1) 查找
        for i, n in enumerate(nums):    # 遍历数组,i 是下标,n 是当前值
            cp = target - n         # 计算需要的另一个数(target 减去当前值)
            if cp in dct:           # 之前遍历过这个数,说明找到配对
                return [dct[cp], i] # 返回「之前那个数的下标, 当前下标」
            dct[n] = i              # 没找到,先把当前值和下标存入哈希表

留言板

加载评论中...
【LeetCode HOT100】15. 三数之和
【LeetCode HOT100】11. 盛最多水的容器
Valaxy v0.28.0-beta.1 驱动|主题-Yunv0.28.0-beta.1