LeetCode-977. 有序数组的平方

最新推荐文章于 2025-01-31 14:37:16 发布

原创最新推荐文章于 2025-01-31 14:37:16 发布 · 135 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #职场和发展

该篇博客介绍了如何解决LeetCode中的第977题，即给定一个非递减排序的整数数组，返回每个数字平方后的数组，依然保持非递减顺序。博主提出使用双指针法，从数组两端开始比较并填充结果数组，确保平方后的值始终有序。这种方法在C++实现中取得了较好的时间和空间效率。

LeetCode-977. 有序数组的平方

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

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

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

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

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

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