LeetCode 热题 HOT 100 —— 1. 两数之和(python解法)
给定一个整数数组 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]
示例2:
输入:nums = [3,3], target = 6
输出:[0,1]
解题思路:
1.暴力求解
由于所求为两数之和,已知列表中的元素,只需要2次遍历就可以找到,这种方法的时间复杂度较高。
代码如下:
class Solution(object):
def twoSum(self, 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]
从图中也可以看出,耗时和内存都是很高的,因此不建议采用。
2.哈希求和(字典)
首先在字典中存储每个数对应的索引,然后用target-这个数,如果差值在字典中存在,那么说明可以相加为target,返回对应索引。
代码如下:
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashmap = {}
for i, num in enumerate(nums):
if target - num in hashmap:
return [hashmap[target - num], i]
hashmap[num] = i
可以看到用时比暴力求解要快很多。