代码随想录算法训练营第2天
977.有序数组的平方
209.长度最小的子数组
59.螺旋矩阵II
leetcode 977.有序数组的平方
题目
977.有序数组的平方
给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。
示例 1：
输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]
解题思路1
使用暴力法，先对原数组中每个数字做平方运算，然后对新数组进行排序
实现代码1
from typing import List
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
# 对原数组中的每个元素进行平方计算
i = 0
while i < len(nums):
nums[i] = nums[i] ** 2
i += 1
# print(nums) # 平方后的新数组
# 对新数组的元素进行递增排序
j = 0
while j < len(nums):
left = 0 # 用于获取位置
right = left + 1 # 用于获取值
while right <= len(nums) - 1:
if nums[right] < nums[left]:
tmp = nums[right]
nums[right] = nums[left]
nums[left] = tmp
left += 1
right += 1
j += 1
return nums
test = Solution()
nums = [-4,-1,0,3,10]
print(test.sortedSquares(nums))
实现代码1在LeetCode执行会超出时间限制，看题解可以直接使用排序函数，如下：
实现代码2
from typing import List
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
# 对原数组中的每个元素进行平方计算
i = 0
while i < len(nums):
nums[i] = nums[i] ** 2
i += 1
# 对新数组的元素进行递增排序
nums.sort()
return nums
解题思路2
使用双指针法：
原数组中有负数，每个元素平方后，数组中的大数一定是在左边或右边
（1）通过左右双指针分别获取数组两边的元素，比较后获得数值大的元素
（2）创建一个空数组，从右边开始存放元素，最后一位是（1）左右指针比较后得到的大数
（3）循环以上操作，左指针向右移动取值，右指针向左移动取值，比较大小，把大数从右向左依次放到结果数组中
（4）当左右指针相遇后，循环结束
实现代码
from typing import List
class Solution:
def sortedSquares(self, nums: List[int]) -> List[int]:
left = 0 # 原数组左边第一位指针
right = len(nums) - 1 # 原数组右边第一位指针
i = len(nums) - 1 # 结果数组右边第一位指针
arr = [float('inf')] * len(nums) # 定义新数组，长度和原数组一样，float('inf')表示正无穷大
while left <= right:
if nums[left] ** 2 > nums[right] ** 2:
arr[i] = nums[left] ** 2
left += 1 # 原数组左指针向右移动
else:
arr[i] = nums[right] ** 2
right -= 1 # 原数组右指针向左移动
i -= 1 # 结果数组指针向左移动
return arr
test = Solution()
nums = [-4,-1,0,3,10]
print(test.sortedSquares(nums))
