刷题第2天：LeetCode977--有序数组的平方--双指针法

Trouville012024-02-27 7:02

给你一个按 非递减顺序 排序的整数数组 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 已按 非递减顺序 排序

思路分析：

**1.**最简单的方法是先平方，然后使用快速排序，这是暴力解法，计算复杂度略高；

2.给出的数组是有序的，只不过包含负数，负数平方之后可能会成为最大的数，所以需要进行判别；

3.数组平方的最大值就在数组的两端，不是最左边就是最右边，不可能是中间，这就可以考虑双指针法了，i指向起始位置，j指向终止位置。再定义一个新数组result，和给的原始数组一样的大小，让k指向result数组终止位置。

python 复制代码

# 双指针法
class Solution:
    def sortedSquares(self, nums:List[int]) -> List[int]:
        left, right, i = 0, len(nums)-1, len(nums)-1
        result = [float('inf)] * len(nums) # 需要提前定义列表，存放结果
        while left <= right:
            if nums[left] ** 2 < nums[right] ** 2:  # 左右边界进行对比，找出最大值
                result[i] = nums[right] ** 2
                right -= 1 # 右指针往左移动
            else:
                result[i] = nums[left]**2
                left += 1  #左指针向右移动
            i -= 1 # 存放结果的指针需要向小的方向平移一位
        return result

总结：本题的关键是给定的数组是有序的，且可以包含负数，则平方后的最大值会出现的数组的两端，采用双指针法比较得出平方后的最大值，并将最大值存入到结果数组中。注意与第1天移除元素中采用的快慢指针相区别。快慢指针的起始下标都是0，而本题则是一个是左指针，一个是右指针，起始点不同。