两数之和
题目描述:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例1:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
在力扣官方的解答中,Python3的解法是基于哈希表的,哈希表同Python的字典一样,也是键值对的形式,所以考虑用字典解题。
思路:
如果是有序升序的nums整数数组,则考虑用到双指针算法,分别指向nums首部和尾部,然后将左右指针所指向的元素和与target进行比较,和<target,左指针向后移动一位,如果和>target,右指针向左移动一位。
此题是无序的nums整数数组,参考官方的解答。创建一个空的字典dict,遍历nums数组中的每一个值nums[i],判断target - nums[i]是否存在于dict中,如果不存在,将nums[i]添加到字典中,否则的话,证明当前的值与存在于dict中的target-nums[i]的和为target,这两个值的索引就是我们最终要返回的结果。
为便于理解:截一个官方的图对照。最终已经添加到哈希表中的6和当前遍历的元素2的索引就是我们需要的返回结果。
代码:
def twoSum(nums, target):
dict = {}
for i, value in enumerate(nums):
if target - value not in dict:
dict[nums[i]] = i
else:
return [dict[target - nums[i]],i]
if __name__ == '__main__':
nums = [2,8,11,15]
target = 26
print(twoSum(nums, target))
运行结果:
[2, 3]