1.两数之和
- 简单
给定一个整数数组 nums
和一个整数目标值 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 <= 103
- -109 <= nums[i] <= 109
- -109 <= target <= 109
- 只会存在一个有效答案
方法一
解题思路:
就是从第一个元素开始依次挨个和后面的元素进行和运算,来查找和为目标数的两个数。第一个没有就再拿第二个,第三个,直至所有元素都已组合过。这就需要进行两层循环。
代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
res = []
for i in range(len(nums)):
for j in range(i+1, len(nums)):
if nums[i] + nums[j] == target:
res.extend([i, j])
break
return res
结果:
分析:
果然两层循环的暴力枚举是占用内存很大的。
-
时间复杂度:O(N^2),其中 N 是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。
-
空间复杂度:O(1)
方法二
解题思路:
用单循环,查找目标值减当前值的差在不在列表里,且索引不能等于当前索引。
代码
class Solution:
def twoSum1(self, nums: List[int], target: int) -> List[int]:
n = len(nums)
res = []
for i in range(n):
k = target - nums[i]
if k in nums:
ind = nums.index(k)
if i == ind:
continue
else:
res.append(i)
res.append(ind)
break
return res
if __name__ == '__main__':
solu = Solution()
a = solu.twoSum1([3, 3], 6)
print(a)
结果:
分析:
一次循环一下子就将时间降下来了。
-
时间复杂度:O(N)
-
空间复杂度:O(1)
方法三
解题思路:
参考了大神们的解法,通过哈希来求解,这里通过字典来模拟哈希查询的过程。
代码:
class Solution:
def twoSum3(self, nums: List[int], target: int) -> List[int]:
hashmap = {}
for i, num in enumerate(nums):
if hashmap.get(target - num) is not None:
return [i, hashmap.get(target - num)]
hashmap[num] = i # 这句不能放在if语句之前,解决list中有重复值或target-num=num的情况
if __name__ == '__main__':
solu = Solution()
a = solu.twoSum3([3, 3], 6)
print(a)
结果:
分析:
这个结果也是快的,只是从另一个角度来解决问题。