给定一个数组和target,求数组中相加的和为target的index,返回一个list。
思路一:
将数组nums排序,然后设定两个指针,一个指针的index为0,往后走;一个指针的index为len(nums)-1,往前走。
若相加数值等于target,满足(返回“nums.index(value)”);
若相加数值小于target,前指针往后走;
若相加数值大于target,后指针往前走。
注意:若两个数的值相等,则需考虑切片找出第2个数的位置,因为index()方法只能list中找到第一个匹配的值。
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
if not nums:
return None
n = sorted(nums) # 复杂度:n*logn
i = 0
j = len(n)-1
plus = 0
res = []
while i<len(n) and j>=0:
plus = n[i]+n[j]
if plus == target:
res.append(nums.index(n[i]))
if n[i]==n[j]: #两个同样的值
res.append(nums[res[0]+1:].index(n[j])+(res[0]+1)) #别忘了加被切掉的那部分
else:
res.append(nums.index(n[j]))
return res
elif plus > target:
j -= 1
elif plus < target:
i += 1
return None
思路二:
使用dict,若i=0,nums[i]=2:
则dict中,key=target-nums[i]=9-2=7,value=i=0,意思即为缺7,则i=0的位置可以满足条件。
class Solution:
def twoSum(self, nums, target):
if len(nums) <= 1:
return False
buff_dict = {}
for i in range(len(nums)):
if nums[i] in buff_dict: #索引key,非value
return [buff_dict[nums[i]], i]
else:
buff_dict[target-nums[i]] = i