开始刷leetcode,记录自己的刷题之路。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
1.两数之和
题目描述+示例
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
input: nums = [2, 7, 11, 15], target = 9
output: [0,1]
解析题干
数组nums特性:元素是整数,数组无序
考虑输入情况:
情况一:数组为空或没有符合要求的
情况二:数组中存在不止一对数值符合要求,例子如下:
eg1:nums = [2, 7, 3, 6, 9, 0], target = 9
// what is the output?
eg2: nums = [2, 7, 11, 15], target = 4
// what is the output?
eg3: nums = [2, 7, 11, 7], target = 9
// what is the output?
题干假设每个输入只会对应一个答案
,因此测试用例不会出现eg1
这种情况。
题干不能重复利用这个数组中同样的元素
,是指eg2
中不能输出[0,0],还是指eg3
中输出[0,1]或[0,3]均可。
思路
对于nums[i]
,在剩余元素中查找target - nums[i]
方法一 暴力求解
从nums[i]之前或之后查找均可
def twoSum(self, nums, target):
if not len(nums):
return []
//nums[i]之后查找
for i in range(len(nums)):
t = target - nums[i]
for j in range(i+1,len(nums)):
if nums[j] == t:
return i,j
//nums[i]之前查找
for i in range(1,len(nums)):
t = target - nums[i]
for j in range(i):
if nums[j] == t:
return j,i
return []
方法二 利用哈希查找
def twoSum(self, nums, target):
if not len(nums):
return []
record = {}
for i in range(len(nums)):
if target - nums[i] in search:
return record.get(target-nums[i]),i
record[nums[i]] = i
return []
总结:查找一个元素有两种方式,一种遍历O(n),一种哈希查找O(1)