LeetCode 1. 两数之和

发表于： 分类于：
字数： 398 阅读：≈ 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]

提示：

  • $2 \leq nums.length \leq 10^4$
  • $-10^9 \leq nums[i] \leq 10^9$
  • $-10^9 \leq target \leq 10^9$
  • 只会存在一个有效答案

解法

思路一：暴力枚举

1
2
3
4
5
6
7
8

class Solution:
    def twoSum(self, nums: list[int], target: int) -> list[int]:
        n = len(nums)
        for i, v1 in enumerate(nums):
            for j in range(i+1, n):
                if v1 + nums[j] == target:
                    return [i, j]
        return [-1, -1]
  • 时间复杂度：$O(n^2)$

思路二：通过哈希表，空间换时间

1
2
3
4
5
6
7
8
9

class Solution:
    def twoSum(self, nums: list[int], target: int) -> list[int]:
        seen = {}
        for i, v1 in enumerate(nums):
            v2 = target - v1
            if v2 in seen:
                return [i, seen[v2]]
            seen[v1] = i
        return [-1, -1]
  • 时间复杂度：$O(n)$
  • 题干中的“可以按任意顺序返回答案”是思路二可行的关键，否则需要额外的结构保存排序