977.有序数组的平方
给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例 1:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
示例 2:
输入:nums = [-7,-3,2,3,11]
输出:[4,9,9,49,121]
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 已按 非递减顺序 排序
进阶:
请你设计时间复杂度为 O(n) 的算法解决本问题
思路:由于平方之后,是个抛物线函数,会打乱之前的顺序,最主要的就是重新排列顺序,只需要左右两个指针,同时往中间走,把平方后大的数放进来,这样就有序了
解法:
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
n = len(nums)
left = 0
right = n - 1
k = n - 1
output = [-1] * n
while (left <= right):
lm = nums[left] ** 2
rm = nums[right] ** 2
if lm < rm:
output[k] = rm
right -= 1
else:
output[k] = lm
left += 1
k -= 1
return output