product-of-array-except-self solution

Author: hyeri0903

product-of-array-except-self/hyeri0903.py

@@ -0,0 +1,21 @@
+class Solution:
+    def productExceptSelf(self, nums: List[int]) -> List[int]:
+        """
+        time complexity : O(n)
+        space complexity : O(n)
+        """
+        n = len(nums)
+        prefix = [1] * n
+        suffix = [1] * n
+        answer = [1] * n
+
+        for i in range(1, n):
+            prefix[i] = prefix[i-1] * nums[i-1]
+
+        for i in range(n-2, -1, -1):
+            suffix[i] = suffix[i+1] * nums[i+1]
+        
+        for i in range(n):
+            answer[i] = prefix[i] * suffix[i]
+        
+        return answer