一、题目
二、解题
1.运用内置函数
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
#内置函数
n = len(nums)
ans = [0] * n
for i in range(n):
ans[i] = nums[i] * nums[i]
ans.sort() # 时复nlogn
return ans
先将每一位平方,再排序。但是sort()函数的时间复杂度为O(nlogn),不满足题目要求。
2.相向双指针
为什么会想到使用这个方法呢?因为题目有一个很重要的点就是nums数组是非递减排序的,也就是不严格的递增。对于有顺序的数组,一定要充分利用有序性。
下面对题目进行分析:
在0两侧的尽头的绝对值最大,越靠近中间的0越小,那么将两个指针指向左右尽头,平方大的就先放入数组,然后移动指针。而平方大实际上就是绝对值大,而绝对值大就是“-负数”和“+正数”比较谁更大。而L指针指向的是负数,R指针指向的是正数,所以判断条件为- nums[l] >= nums[r]。
而值为0的时候是一个特殊的存在,当指针所指的值为0时,就不需要再比较两指针的绝对值大小了。因为剩下的数值里面是符号相同的数,从指向0的指针到另一个指针的值的绝对值必然是增加的(因为nums数组是递增的,充分利用有序性)。如若剩下的数组为【-4,-3,-2,-1,0】,此时即L指向-4,R指向0,那么从0到-4的绝对值必然是增加的。然后我们将这段数组的值直接存入答案数组即可。注意这里要从远离0的位置开始存,因为我们之前一直是将当前的绝对值大值存入数组,这里也是一样。
另外,博主在写代码的时候,先是将ans数组(答案数组)令为空,然后使用insert()函数在列表的头位(下标为0的位)插入的数值(要求非递减顺序),但是运行起来发现时间太慢了,所以改为了idx存需要放的下标,再赋值。原代码为注释的代码。
代码如下:
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
# 双指针
n = len(nums)
# ans = []
ans = [0] * n
idx = n-1
l,r = 0,n-1 #指针
while l <= r:
if nums[l] == 0:
for i in range(r-l+1):
# ans.insert(0,nums[r-i] ** 2)
ans[idx] = nums[r-i] ** 2
idx -= 1
break
if nums[r] == 0:
for i in range(r-l+1):
# ans.insert(0,nums[l+i] ** 2)
ans[idx] = nums[l+i] ** 2
idx -= 1
break
if - nums[l] >= nums[r]:
# ans.insert(0,nums[l] ** 2)
ans[idx] = nums[l] ** 2
idx -= 1
l += 1
else:
# ans.insert(0,nums[r] ** 2)
ans[idx] = nums[r] ** 2
idx -= 1
r -= 1
return ans
感谢你看到这里!一起加油吧!