fqy131314

有序数组的平方(数组篇)

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

思路:

        ①暴力排序

        ②双指针法

①暴力排序

        每个数平方之后,排个序

        时间复杂度为O(n+nlogn)

代码:

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        for (int i = 0; i < A.size(); i++) {
            A[i] *= A[i];
        }
        sort(A.begin(), A.end()); // 快速排序
        return A;
    }
};

②双指针法

        数组其实是有序的, 只不过负数平方之后可能成为最大数了。

        那么数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。

        此时可以考虑双指针法了,i指向起始位置,j指向终止位置。

        定义一个新数组result,和A数组一样的大小,让k指向result数组终止位置。

        如果A[i] * A[i] < A[j] * A[j] 那么result[k--] = A[j] * A[j]; 。

        如果A[i] * A[i] >= A[j] * A[j] 那么result[k--] = A[i] * A[i]; 。

代码:

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int k = nums.size() - 1;
        vector<int> result(nums.size(),0);

        for(int i = 0,j = nums.size() - 1; i <= j; )
        {
            if(nums[i] * nums[i] < nums[j] * nums[j])
            {
                result[k] = nums[j] * nums[j];
                j--;
                k--; 
            }
            else
            {
                result[k] = nums[i] * nums[i];
                i++;
                k--;
            }
        }

        return result;
    }
};

posted on 2023-04-17 14:34  会飞的鱼-blog  阅读(23)  评论(0)    收藏  举报  来源

导航