977. Squares of a Sorted Array

Задача

Дан отсортированный массив nums. Верните массив квадратов элементов в неубывающем порядке.

Примеры

Пример 1

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

Пояснение

После возведения в квадрат: [16,1,0,9,100]; после сортировки: [0,1,9,16,100].

Пример 2

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

Решение

Решение

/**
 * Time Complexity: O(n) - один проход по массиву
 * Space Complexity: O(1) - не считая выходной массив, используем только константное дополнительное пространство
 */
var sortedSquares = function(nums) {
    let left = 0;
    let right = nums.length - 1;
    const result = new Array(nums.length);
    let position = nums.length - 1;
 
    while (left <= right) {
        const leftSq = nums[left] ** 2;
        const rightSq = nums[right] ** 2;
 
        if (leftSq <= rightSq) {
            result[position] = rightSq;
            right--;
        } else {
            result[position] = leftSq;
            left++;
        }
        position--;
    }
 
    return result;
};