two-sumsV2.py (384B)
1 class Solution: 2 def twoSum(self, nums: List[int], target: int) -> List[int]: 3 4 num_dict = {} 5 6 for i in range(len(nums)): 7 num_dict[nums[i]] = i 8 9 for i in range(len(nums)): 10 current = nums[i] 11 if (target - current) in num_dict and num_dict[target - current] != i: 12 return [i, num_dict[target-current]]