给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例 1:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
left = 0
right = len(nums)-1
pos = len(nums)-1
res = [0 for _ in range(len(nums))]
while left <= right:
if nums[left]*nums[left] > nums[right]*nums[right]:
res[pos] = nums[left]*nums[left]
left += 1
else:
res[pos] = nums[right]*nums[right]
right -= 1
pos-=1
return res