algorithms

Algorithm implementations
git clone git://git.laack.co/algorithms.git
Log | Files | Refs | README

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]]