diff --git a/two-sum/Bumsu-Yi.py b/two-sum/Bumsu-Yi.py index b71026ab2..6b17fdde2 100644 --- a/two-sum/Bumsu-Yi.py +++ b/two-sum/Bumsu-Yi.py @@ -1,5 +1,14 @@ """ https://leetcode.com/problems/two-sum/ """ -# +class Solution: + def twoSum(self, nums: List[int], target: int) -> List[int]: + numMap = {} + + for i in range(len(nums)): + complement = target - nums[i] + if complement in numMap: + return [numMap[complement], i] + + numMap[nums[i]] = i