【20190520】【每天一道算法题】有序数组的平方（双指针）

问题

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

---

思路及代码

1. Python实现

class Solution:
    def sortedSquares(self, A: List[int]) -> List[int]:
        # # 方法一：求数组元素的平方，然后排序平方后的数组
        # return sorted(x**2 for x in A)
        
        # 方法二：（双指针法）从原数组两边比较元素的绝对值，绝对值较大的数先计算平方并放入输出的新数组中
        left, right = 0, len(A)-1
        result = []
        while left != right:
            if abs(A[left]) >= abs(A[right]):
                result.append(A[left]**2)
                left += 1
            else:
                result.append(A[right]**2)
                right -= 1
        result.append(A[left]**2)
        return result[: : -1]
                

---

知识点

1. 字符串切片：反向输出是Array[: : -1]。

2. Python求绝对值用abs(num)