977-squares-of-a-sorted-array.py

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        res = []
        left, right = 0, len(nums) - 1
        while left <= right:
            if nums[left] ** 2 < nums[right] ** 2:
                res.append(nums[right] ** 2)
                right -= 1
            else:
                res.append(nums[left] ** 2)
                left += 1
        return res[:: - 1]
        
# time O(n)
# space O(n)
# using array and two pointers opposite direction and shrink type