leetcode每日一题: 977. 有序数组的平方

leetcode每日一题: 977. 有序数组的平方

题目描述

给定一个按非递减顺序排序的整数数组 A，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。

示例 1：

输入：[-4,-1,0,3,10]
输出：[0,1,9,16,100]

示例 2：

输入：[-7,-3,2,3,11]
输出：[4,9,9,49,121]

提示：

  1. 1 <= A.length <= 10000
  2. -10000 <= A[i] <= 10000
  3. A 已按非递减顺序排序。

通过次数78,914提交次数107,279

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

方法一: 这很pythonic

python嘛, 写的代码当然要pythonic一点嘿嘿嘿

class Solution:
    def sortedSquares(self, A: List[int]) -> List[int]:
        return sorted(list(map(lambda x:x**2,A)))

方法二: 双指针法

如果数组 A 中的所有数都是非负数，那么将每个数平方后，数组仍然保持升序；如果数组 AA 中的所有数都是负数，那么将每个数平方后，数组会保持降序。

这样一来，如果我们能够找到数组 A 中负数与非负数的分界线，那么就可以用类似「归并排序」的方法了。具体地，我们设 neg 为数组 A 中负数与非负数的分界线，也就是说，A[0]A[0] 到 A[neg]A[neg] 均为负数，而 A[neg+1]A[neg+1] 到 A[n-1]A[n−1] 均为非负数。当我们将数组 A 中的数平方后，那么 A[0]A[0] 到 A[neg]A[neg] 单调递减，A[neg+1]A[neg+1] 到 A[n-1]A[n−1] 单调递增。

由于我们得到了两个已经有序的子数组，因此就可以使用归并的方法进行排序了。具体地，使用两个指针分别指向位置 neg 和 neg+1，每次比较两个指针对应的数，选择较小的那个放入答案并移动指针。当某一指针移至边界时，将另一指针还未遍历到的数依次放入答案。

class Solution:
    def sortedSquares(self, A: List[int]) -> List[int]:
        n = len(A)
        negative = -1
        for i, num in enumerate(A):
            if num < 0:
                negative = i
            else:
                break
        
        i, j = negative, negative+1
        ans = []
        while i>=0 or j<n:
            if i<0:
                ans.append(A[j]*A[j])
                j += 1
            elif j == n:
                ans.append(A[i]*A[i])
                i -= 1
            elif A[i] * A[i] < A[j] * A[j]:
                ans.append(A[i]*A[i])
                i -= 1
            else:
                ans.append(A[j]*A[j])
                j += 1
        
        return ans

感谢观看