给你一个按 非递减顺序 排序的整数数组 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) 的算法解决本问题
#最直观的思路
直接先将数组中的所有数进行排序,
随后再应用一个排序即可
#双指针思路
这到题可以利用双指针的思想
首先令左右指针分别指向数组的头部和尾部,
同时我们在创建一个指针来记录当前结果的存放位置。
然后开始对比首尾两个数字的绝对值,
取较大的一个平方后存放在ans[position]处,
也就是将最大的数存放在列表的末尾,
随后将position-1,以方便存放下一个结果,这样存起来的结果就是排好序的。
重复此过程,直至left> right
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
# for i in range(len(nums)):
# nums[i] *= nums[i]
# return sorted(nums)
left, right, position = 0, len(nums) -1, len(nums) - 1
ans= [0] * len(nums)
while left <= right:
if abs(nums[left]) > abs(nums[right]):
ans[position] = nums[left] ** 2
left += 1
else:
ans[position] = nums[right] ** 2
right -= 1
position -= 1
return ans