注:leetcode会自动补齐头文件、不需要考虑参数传入问题,只需要我们完成主函数的书写
https://blog.csdn.net/qq_43338695/article/details/102508769
1. 题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,3], target = 6
输出:[0,1]
https://leetcode-cn.com/problems/two-sum/
2. python解决方法
2.1 两层循环暴力法(时间复杂度O(n^2))
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i in range(len(nums)-1):
for j in range(i+1,len(nums)):
if nums[i]+nums[j]==target:
return [i,j]
2.2 哈希表
利用dict(就是python哈希表的例子)查找target-num的时间复杂度O(1)提速
https://www.cnblogs.com/kumata/p/9157875.html
2.2.1 先创建字典,后搜索
class Solution():
def twoSum(self, nums: List[int], target: int) -> List[int]:
map={}
for i in range(len(nums)):
# 构建字典的时候,如果出现两个相同的key后面的会覆盖,比如数组[3,3]返回{3: 1}
map[nums[i]]=i
for i in range(len(nums)):
# 因为这里是在nums里面循环,所以对于相同数组成target的情况没有影响,比如[3,3],i记录到的是1,差值搜索到的3这时候是后一个3,也就是索引是2
if target-nums[i] in map and i!=map[target-nums[i]]:
# 这里要考虑不能使用同一个元素
return [i,map[target-nums[i]]]
else:
continue
2.2.2 边往表里添加键值,边向前搜索
class Solution():
def twoSum(self, nums: List[int], target: int) -> List[int]:
map={}
for i ,num in enumerate(nums):
if target-num in map:
return [map[target-num],i]
else:
map[num]=i
ps:还没有数据结构的基础,刷题记录,有问题希望大佬们指正