给定一个整数数组 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]
1.暴力破解法
暴力破解法的思想在于遍历两遍数组。值得注意的是第二层循环不要再从0开始,而是从第一层循环标签 i 的下一位开始。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
lens = len(nums)
for i in range(lens-1):
for j in range(i+1, lens):
if nums[i]+nums[j] == target:
return [i, j]
2.列表法
首先用目标值target与数组nums[i]的值做减法得到数值a,再判断a是否在数组nums中。如果a在nums中,则通过nums.index(a)可以得到a在nums中的数组下标;若a不在nums中,则继续循环。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
lens = len(nums)
for i in range(lens):
a = target - nums[i]
if a in nums:
b = nums.index(a)
if b == i:
continue
else:
return [i, b]
else:
continue
3.字典法
字典法很巧妙,建议大家在纸上写一写整个流程。
思想在于将nums[i]的值作为字典的key,将i的值作为字典的value。开始循环时,用target-nums[i]的值与字典中的key作比较,若target-nums[i]已经在字典中,则返回target-nums[i]所对应的value和循环次数i;若不在字典中,则继续用nums[i]和i建立字典。
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
d = {}
lens = len(nums)
for i in range(lens):
a = target - nums[i]
if a in d:
return[d[a], i]
else:
d[nums[i]] = i