leetcode-两数之和
1.题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
示例
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
2. 解题思路
a. 暴力法
枚举数组中每一个元素\(x\),判断是否存在 \(target-x\).
def twoSum1(self, nums: List[int], target: int) -> List[int]:
n = len(nums)
for i in range(n-1):
for j in range(i+1,n):
if nums[i]+nums[j]==target:
return [i,j]
return []
b. 两遍哈希
第一次遍历,将元素的值和索引加入哈希表中(字典),
第二次遍历,判断\(target-x\)是否存在与哈希表中。
注意:目标元素不能是x本身。
def twoSum2(self, nums: List[int], target: int) -> List[int]:
n = len(nums)
hash = {}
for i in range(n):
hash[nums[i]]=i
for j in range(n):
complement = target - nums[j]
if complement in hash and j!=hash[complement]:
return [j,hash[complement]]
return []
c. 一遍哈希
针对方法二可以优化,将两次遍历合并到一次。对于每一个x, 首先判断哈希表中是否存在\(target-x\), 然后将x插入到哈希表中。
def twoSum(self, nums: List[int], target: int) -> List[int]:
hashtable = dict()
for i, num in enumerate(nums):
if target - num in hashtable:
return [hashtable[target - num], i]
hashtable[nums[i]] = i
return []