leetCode第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 <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
第一种方法就是暴力枚举
先放置两个指针,第二个指针置于第一个之后,然后逐一移动第二个指针直至遍历完成,之后就把第一个指针移动一位,第二个指针置于第一个指针之后,重复执行。
# python3
class Solution:
def twoSum(self, nums: [int], target: int) -> [int]:
result = []
for i in range(len(nums)):
for j in range(i+1,len(nums)):
if nums[i]+nums[j] == target:
result.append(i)
result.append(j)
return result
return result
提交后发现暴力法的效果并不是很好,应该要进行优化
在暴力法中,因为循环的缘故,导致我们是在不断的重复已经扫描过的值,这浪费了大量的时间.
因此我们可以引入hash map用来存储下标和对应的值(在python中可以用字典实现hash map的功能)
先设立一个指针指向第一个元素,此时hash map应该是个空表
先指向4,目标值为20,我们就往hash map里找是否有16这个值,但没找到(此时为空表),没有找到就把4这个元素放入hash map中,然后指针后移。
当指针指向7这个数时,在hash map里可以找到13这个值,于是利用hash map里13对应的下标和指针所在下标返回
# python3
class Solution:
def twoSum(self, nums: [int], target: int) -> [int]:
hash_map = {} # 创建hash map,用字典实现
result = []
for i in range(len(nums)):
index = hash_map.get(target-nums[i])
if None != index : # hash map中有,则记录下标
result.append(index)
result.append(i)
break
else: # hash map中没有对应元素,则将新元素存入hash map
hash_map[nums[i]] = i
return result
在整个执行过程中,我们只对nums扫描了一遍,因此是一个线性的时间复杂度,为O(m)
内存占用变大了不少,但时间上缩短了很多。