leetcode 977. Squares of a Sorted Array

Given an array of integers A sorted in non-decreasing order, return an array of the squares of each number, also in sorted non-decreasing order.

 

Example 1:

Input: [-4,-1,0,3,10]
Output: [0,1,9,16,100]

Example 2:

Input: [-7,-3,2,3,11]
Output: [4,9,9,49,121]

 

Note:

1. 1 <= A.length <= 10000
2. -10000 <= A[i] <= 10000
3. A is sorted in non-decreasing order.

题目大意：输入一个非递减序的数组，返回数组元素的平方，要求非递减序。

 

思路一：对每个元素平方操作，然后排序。

时间复杂度：O(nlogn),

空间复杂度：O(n)

 

思路二：考虑原数组是有序的，找到正负数的分解点，两个指针向左向右，比较。

vector<int> sortedSquares(vector<int>& A) {
        vector<int> ans(A.size(), 0);
        int i = 0, len = A.size() - 1, k = 0;
        
        while (i < A.size() && A[i] < 0) ++i;
        
        int j = i - 1;
        while (j >= 0 && i <= len) {
            if (-A[j] < A[i]) {
                ans[k++] = A[j] * A[j];
                j--;
            } else {
                ans[k++] = A[i] * A[i];
                i++;
            }
        }
        while (j >= 0) {
            ans[k++] = A[j] * A[j];
            j--;
        }
        while (i <= len) {
            ans[k++] = A[i] * A[i];
            i++;
        }
        return ans;
    }

或者直接从两端开始比较，此时大的数的平方肯定是最大的。

vector<int> sortedSquares(vector<int>& A) {
        vector<int> ans(A.size(), 0);
        int i = 0, j = A.size() - 1, k = A.size() - 1;
        while (i <= j) {
            if (abs(A[i]) < abs(A[j])) {
                ans[k--] = A[j] * A[j];
                --j;
            } else {
                ans[k--] = A[i] * A[i];
                ++i;
            }
        }
        return ans;
    }