977. 有序数组的平方正负数平方排序双指针

最新推荐文章于 2024-09-14 14:36:23 发布

原创最新推荐文章于 2024-09-14 14:36:23 发布 · 606 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种算法，用于处理一个按非递减顺序排序的整数数组，将数组中的每个数字进行平方操作，并确保结果数组同样按非递减顺序排序。通过双指针技巧，该算法实现了高效处理，避免了额外的排序步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

示例 1：

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

示例 2：

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

提示：

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

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        int N=A.size();
        vector<int> res(N);

        if(A[0]>=0)
        {
            for(int i=0;i<N;i++)
                res[i]=A[i]*A[i];
            return res;
        }
        int i=N-1,j=0,t=N-1;
        while(i>=j)
        {
            int ai=A[i]*A[i];
            int aj=A[j]*A[j];
            if(ai<aj)
            {
                res[t--]=aj;
                j++;
            }
            else
            {
                res[t--]=ai;
                i--;
            }
        }
        return res;
    }
};