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]
出版作答(python3):
class Solution:def sortedSquares(self, nums: List[int]) -> List[int]:nums_sq=[]n=0for i in nums:j=i*inums_sq.append(j)n = len(nums_sq)for i in range(n):for j in range(0, n - i - 1):if nums_sq[j] > nums_sq[j + 1]:nums_sq[j], nums_sq[j + 1] = nums_sq[j + 1], nums_sq[j]return nums_sq
提交的时候超出时间限制。先平方,之后采用手动的冒泡排序,超时。
第二版:
class Solution:def sortedSquares(self, nums: List[int]) -> List[int]:nums_sq=[]n=0for i in nums:j=i*inums_sq.append(j)nums_sq.sort()return nums_sq
题目允许调用函数,可以不手写排序,节省时间。
最优版:
class Solution:def sortedSquares(self, nums: List[int]) -> List[int]:n = len(nums)result = [0] * nleft, right = 0, n - 1pos = n - 1while left <= right:if abs(nums[left]) > abs(nums[right]):result[pos] = nums[left] ** 2left += 1else:result[pos] = nums[right] ** 2right -= 1pos -= 1return result
双指针法:
双指针从两端找平方最大值,逆序填入新数组。
1、为何逆序插入?
我们从数组两端找最大平方值,大的数应该放在结果数组的最后面,所以我们需要从后往前插入。
最大平方值一定出现在原数组的两端。设置双指针从两端开始比较
创建一个结果数组 result,长度相同,全部初始化为 0。—>result=[0]*n
大致思路:
比较两端绝对值大小:
谁的绝对值大,说明平方后也更大;
将较大的平方放入结果数组的最后一个位置;
创建左右指针:left 指向最左,right 指向最右;移动对应指针(左边大就 left += 1,右边大就 right -= 1);
重复上述过程,直到左右指针相遇。