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

LeetCode977题:

给你一个按 非递减顺序 排序的整数数组 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数组终止位置。

# 双指针法
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,而本题则是一个是左指针,一个是右指针,起始点不同。

你可能感兴趣的:(算法,数据结构,leetcode)