两数之和
给定一个整数列表 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
样例1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
输入
>2,7,11,15
9
输出
[0,1]
因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
样例2:
输入:nums = [1,2,4], target = 6
输出:[1,2]
代码:
arr = input("输入列元素以\",\"分隔:")
target =int(input("输入目标值:"))
nums = [int(n) for n in arr.split(",")]
class Solution(object):
def twoSum(self,nums,target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
dict = {}
for i in range(len(nums)):
if target - nums[i] not in dict:
dict[nums[i]] = i
else:
return [dict[target - nums[i]],i]
a=Solution()
b=a.twoSum(nums,target)
print(b)
|