LeetCode刷题之路:977. 有序数组的平方

如果帮助到您,还请点个关注吧,hahaha

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

进阶:

请你设计时间复杂度为 O(n) 的算法解决本问题

#最直观的思路

直接先将数组中的所有数进行排序,
随后再应用一个排序即可

#双指针思路

这到题可以利用双指针的思想
首先令左右指针分别指向数组的头部和尾部,
同时我们在创建一个指针来记录当前结果的存放位置。
然后开始对比首尾两个数字的绝对值,
取较大的一个平方后存放在ans[position]处,
也就是将最大的数存放在列表的末尾,
随后将position-1,以方便存放下一个结果,这样存起来的结果就是排好序的。
重复此过程,直至left> right

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        # for i in range(len(nums)):
        #     nums[i] *= nums[i]
        # return sorted(nums)
        left, right, position = 0, len(nums) -1, len(nums) - 1
        ans= [0] * len(nums)
        while left <= right:
            if abs(nums[left]) > abs(nums[right]):
                ans[position] = nums[left] ** 2
                left += 1
            else:
                ans[position] = nums[right] ** 2
                right -= 1
            position -= 1
        return ans

你可能感兴趣的:(力扣LeetCode,算法,leetcode,数据结构,python,有序数组的平方)