Day-4 Squares of a sorted array

The below solution is to the problem present in Leetcode's Introduction to Data Structures Array-101 part.

Problem Statement

Given an array of integers A sorted in non-decreasing order, return an array of the squares of each number, also in sorted non-decreasing order.

Example 1

Input: [-4,-1,0,3,10] Output: [0,1,9,16,100] 

Example 2

Input: [-7,-3,2,3,11] Output: [4,9,9,49,121] 

Solution Approach

1. Iterate over the list.
2. Calculate the square of each element.
3. Append the squares of each element to a new list.
4. Sort and return the new list.

class Solution: def sortedSquares(self, A: List[int]) -> List[int]: square_elements = [] for item in A: prod = item*item square_elements.append(prod) square_elements.sort() return square_elements 

NOTE - Leetcode has a skeleton structure for codes that uses classes.