题解 | 两数之和python

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param nums int整型一维数组 
# @param target int整型 
# @return int整型一维数组
#
class Solution:
    def twoSum(self , nums: List[int], target: int) -> List[int]:
        # write code here
        my_dict  = {}
        for i in range(len(nums)):
            if (target - nums[i]) in my_dict:
                return [my_dict[target-nums[i]]+1,i+1]
            else:
                my_dict[nums[i]] = i