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

fromtypingimportList

classSolution:
defsortedSquares(self, A: List[int]) ->List[int]:
B=sorted(A, key=abs) # sort by absolute values
res=list()
foreleminB:
res.append(elem*elem)
returnres

# below is testing 
sol=Solution()
print(sol.sortedSquares([-4,-2, 3,6]))