题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解题思路一
双指针遍历。对原列表进行排序并且记录每个元素排序前的索引,然后left指针从左到右移动,right从右到左移动,如果left位置和right重合,直接返回,如果left指针对应的值加上right指针对应的值等于taget,返回left指针、right指针对应的值在排序前列表中的索引。
(1)记录列表中每个元素及其索引
nums_index = [(v,index) for index,v in enumerate(nums)]
(2)排序
nums_index.sort()
(3)返回及指针更新条件判断
while left < right:
curr = nums_index[left][0] + nums_index[right][0]
if curr == target:
return [nums_index[left][1],nums_index[right][1]]
elif curr < target:
left += 1
else:
right -= 1
完整代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
nums_index = [(v,index) for index,v in enumerate(nums)]
nums_index.sort()
left = 0
right = len(nums) -1
while left < right:
curr = nums_index[left][0] + nums_index[right][0]
if curr == target:
return [nums_index[left][1],nums_index[right][1]]
elif curr < target:
left += 1
else:
right -= 1
解题思路二
字典查找。思想跟解题思路一类似,只不过是把列表中的元素值和索引放到了字典中。
遍历整个列表,target值减去当前元素的值,如果这个数是字典的key,则输出当前元素的索引和这个数在字典中的value。
字典也叫哈希表,好处是可以实现快速查找,时间复杂读O(1)
直接上代码,没什么好说的。
完整代码
class Solution:
def twoSum(self,nums:List[int],target:int) -> List[int]:
hashmap = {}
for i in range(len(nums)):
another_num = target - nums[i]
if another_num in hashmap:
return [i,hashmap[another_num]]
hashmap[nums[i]] = i
return None
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum