Update Squares of a Sorted Array - Leetcode 977.py

Author: gahogg

Squares of a Sorted Array - Leetcode 977/Squares of a Sorted Array - Leetcode 977.py

@@ -34,3 +34,26 @@ def sortedSquares(self, nums: List[int]) -> List[int]:
 
 # Time Complexity: O(n)
 # Space Complexity: O(n)
+
+# Optimal Solution for Bootcamp
+class Solution:
+    def sortedSquares(self, nums: List[int]) -> List[int]:
+        n = len(nums)
+        L, R = 0, n-1
+        result = []
+
+        for i in range(n):
+            nums[i] = nums[i] ** 2
+
+        while L <= R:
+            if nums[L] > nums[R]:
+                result.append(nums[L])
+                L += 1
+            else:
+                result.append(nums[R])
+                R -= 1
+
+        result.reverse()
+        return result
+# Time: O(n)
+# Space: O(n)