Leetcode 977 有序数组的平方

输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]
示例 2：

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

提示：

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 已按 非递减顺序 排序
 

进阶：

请你设计时间复杂度为 O(n) 的算法解决本问题。

题目分析

双指针的做法基于一个简单的事实：一个非递减的、正数负数均有可能的数组中，元素平方最大值既可能存在于数组头部也可能存在于数组尾部。

所以维护两个指针i,j分别指向数组头部和尾部，每次返回一个较大值，然后移动较大值对应的指针。

AC代码

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

总结