代码随想录：977. 有序数组的平方

最新推荐文章于 2024-09-13 16:27:15 发布

_Power_Y

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量157

点赞数 2

分类专栏：力扣文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_74924951/article/details/141899072

版权

力扣专栏收录该内容

43 篇文章 0 订阅

订阅专栏

977. 有序数组的平方

法一：用函数sort来写

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
     for(auto &i:nums)i*=i;
     sort(nums.begin(),nums.end());
     return nums;
    }
};

法二：双指针

因为原数组有序，所以我们用双指针遍历两边，把大的平方放入答案里面即可

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int n = nums.size();
        vector<int> ant(n);
        for (int i = 0, j = n - 1, k = n - 1; i <= j; k--) {
            if (nums[i] * nums[i] > nums[j] * nums[j]) {
                ant[k] = nums[i] * nums[i];
                i++;
            } else {
                ant[k] = nums[j] * nums[j];
                j--;
            }
        }
        return ant;
    }
};