solved: two_sum using a dictionary and algebra complement idea. -@iam…

…serda
iamserda · Feb 25, 2025 · 1ec9296 · 1ec9296
1 parent 94768a1
commit 1ec9296
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/neetcodeio/algostructybeginners/Lv1-Arrays/two_sum_with_dict.py b/neetcodeio/algostructybeginners/Lv1-Arrays/two_sum_with_dict.py
@@ -0,0 +1,14 @@
+class Solution:
+    def two_sum(self, nums: list[int], target: int) -> list[int]:
+        complements = {}
+        for index, num in enumerate(nums):
+            complements[num] = index
+            complement = target - num
+            if complement in complements:
+                return [complements[complement], index]
+
+
+arr = [3, 4, 5, 6]
+target = 7
+sol = Solution()
+assert sol.two_sum(arr, target) == [0, 1]  # efficient solution, O(n) time, O(n) space