题目:给定一个整数数组
nums
和一个整数目标值target
,请你在该数组中找出 和为目标值target
的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
这题比较简单,用的方法很容易就能解答,现在直接上
的方法。
方法一:
利用一个元素为元组的列表存储每个数以及对应的索引,然后对该列表排序,该过程是NlogN的复杂度,最后用双指针的思路对列表的元素进行判断即可。
def twoSum(self, nums: List[int], target: int) -> List[int]:
if nums[0] + nums[1] == target:
return [0, 1]
nums = [(v, i) for i, v in enumerate(nums)]
nums.sort()
left, right = 0, len(nums) - 1
while left < right:
if nums[left][0] + nums[right][0] == target:
return [nums[left][1], nums[right][1]]
elif nums[left][0] + nums[right][0] < target:
left += 1
else:
right -= 1
方法二:
该方法更牛逼,是O(N)的复杂度,但是有点不好理解有点绕。
def twoSum(self, nums: List[int], target: int) -> List[int]:
if nums[0] + nums[1] == target:
return [0, 1]
record = {}
for i, v in enumerate(nums):
# 要和当前数相加等于target的数为diff,如果后续元素中有diff,那么就是要输出的数
diff = target - v
# 如果当前元素在record中,意味着当前元素能和曾经出现过的相加等于target,故输出
if v in record:
# 当某个v存在于record中,意味着该v是曾经的某个diff,而当时那个diff对应着当时与diff相加满足条件的那个v,我们在当时记录了那个v的索引,因此这里为record[v]
return [record[v], i]
else:
# 能使和diff相加为target的当前元素的下标为i
record[diff] = i