1.题目介绍(力扣序号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 <= nums.length <= 10^4
-10^9 <= nums[i] <= 10^9
-10^9 <= target <= 10^9
只会存在一个有效答案
**进阶:**你可以想出一个时间复杂度小于 O(n2) 的算法吗?
解法一
暴力法(时间复杂度 O(n^2))
基本思路:
- 使用两层循环遍历数组中的所有可能的两数组合。
- 在内层循环中,检查这两个数的和是否等于目标值。
- 如果找到满足条件的组合,返回这两个数的下标。
def two_sum_brute_force(nums, target):
for i in range(len(nums)):
for j in range(i + 1, len(nums)):
if nums[i] + nums[j] == target:
return [i, j]
return []
解法二
哈希表法(时间复杂度 O(n))
基本思路:
- 创建一个哈希表,用于存储数组中每个数的值和其对应的下标。
- 遍历数组,对于每个数,计算其与目标值的差值(即所需的另一个数)。
- 检查这个差值是否已经存在于哈希表中。
- 如果存在,说明找到了这两个数,返回它们的下标。
def two_sum_hash_map(nums, target):
num_to_index = {}
for index, num in enumerate(nums):
complement = target - num
if complement in num_to_index:
return [num_to_index[complement], index]
num_to_index[num] = index
return []
解法三
双指针法(先排序,时间复杂度 O(n log n))
基本思路:
- 先将数组元素和对应的下标一起排序。
- 使用两个指针,一个指向数组的开头,一个指向数组的末尾。
- 计算两个指针指向的数的和,根据和与目标值的比较,调整指针的位置:
- 如果和等于目标值,返回这两个数的下标。
- 如果和小于目标值,移动左指针向右。
- 如果和大于目标值,移动右指针向左。
def two_sum_two_pointers(nums, target):
nums_with_index = list(enumerate(nums))
nums_with_index.sort(key=lambda x: x[1])
left, right = 0, len(nums) - 1
while left < right:
current_sum = nums_with_index[left][1] + nums_with_index[right][1]
if current_sum == target:
return [nums_with_index[left][0], nums_with_index[right][0]]
elif current_sum < target:
left += 1
else:
right -= 1
return []