给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍
来源:力扣(LeetCode)
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
hashmap={}
for index,num in enumerate (nums):
temp = target - num
if temp in hashmap:
return [hashmap[temp],index]
hashmap[num]=index
return None
if __name__ == '__main__':
nums = [int(i) for i in input("Please input your number array:").split()]
target = int(input("and your target number:"))
s = Solution().twoSum(nums,target)
print("return your number index:",end="")
print(s)