两数之和
题目:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下
标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路
- 遍历nums的每个元素,检查target减去该元素的值是否在nums里面
- 注意避免target正好是某元素的两倍的情况
代码
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
for i, num in enumerate(nums): # 遍历列表获取索引和数值
dev = target - num # 算出另一个元素的匹配值
if dev != num and dev in nums: # 如果该匹配值不等于num且在nums列表里存在,匹配成功
return [i, nums.index(dev)]
elif dev == num and nums.count(dev) >= 2: # 如果两个元素相等,那么只要nums里面该元素不止1个,即可匹配成功
nums.pop(i) # 为了找到第二个匹配元素,先把第一个剔除
return [i, nums.index(dev) + 1] # 使用.index()方法找到第二个元素的索引,注意该索引是剔除了第一个元素后的索引,所以需要+1
return -1